RandomizeAction Struct Reference

Collaboration diagram for RandomizeAction:

Public Member Functions
void	run (TensorContext &ctx) override
Public Member Functions inherited from nnkit::Action
virtual	~Action ()=default

Detailed Description

Definition at line 26 of file Randomize.cpp.

Member Function Documentation

run()

void RandomizeAction::run ( TensorContext &  ctx )

inlineoverridevirtual

Implements nnkit::Action.

Definition at line 28 of file Randomize.cpp.

  {
    int seed = std::chrono::system_clock::now().time_since_epoch().count();
 
    std::minstd_rand rand(seed);
    std::normal_distribution<float> dist(0.0f, 2.0f);
 
    for (uint32_t n = 0; n < ctx.size(); ++n)
    {
      using nncc::core::ADT::tensor::Accessor;
 
      auto fn = [&dist, &rand](const TensorContext &ctx, uint32_t n, Accessor<float> &t) {
        using nncc::core::ADT::tensor::Index;
        using nncc::core::ADT::tensor::IndexEnumerator;
 
        for (IndexEnumerator e{ctx.shape(n)}; e.valid(); e.advance())
        {
          t.at(e.current()) = dist(rand);
        }
      };
 
      ctx.getMutableFloatTensor(n, fn);
    }
  }

References nnkit::TensorContext::size().

Referenced by package.infer.session::inference().

---

The documentation for this struct was generated from the following file:

• compiler/nnkit/actions/builtin/Randomize.cpp