Method shape

compiler/circle-eval-diff/src/Tensor.h:47–56

  loco::Dimension &dim(uint32_t axis) { return _dims.at(axis); }

  void shape(std::initializer_list<uint32_t> dims)
  {
    rank(dims.size());

    uint32_t axis = 0;
    for (auto d : dims)
    {
      dim(axis++) = d;
    }
  }

private:
  std::vector<loco::Dimension> _dims;