Caffe2 - C++ API
A deep learning, cross platform ML framework
Public Member Functions | Data Fields
at::DimCounter Struct Reference

Public Member Functions

 DimCounter (IntArrayRef shape, Range range)
void increment (const std::array< int64_t, 2 > &step)
bool is_done () const
std::array< int64_t, 2 > max_2d_step () const

Data Fields

IntArrayRef shape
Range range
DimVector values
int64_t offset

Detailed Description

Definition at line 55 of file TensorIterator.h.

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