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:

• aten/src/ATen/native/TensorIterator.h
• aten/src/ATen/native/TensorIterator.cpp