Mila::Dnn::AxisPartition Struct Reference

Information about axis partitioning of a tensor. More...

Public Attributes
int64_t	axis_size
	Size of the axis dimension.
int64_t	inner_size
	Product of dimensions after axis.
int64_t	normalized_axis
	Axis normalized to [0, ndim).
int64_t	num_slices
	outer_size * inner_size (for statistics buffers)
int64_t	outer_size
	Product of dimensions before axis.

Detailed Description

Information about axis partitioning of a tensor.

Member Data Documentation

axis_size

int64_t Mila::Dnn::AxisPartition::axis_size

Size of the axis dimension.

inner_size

int64_t Mila::Dnn::AxisPartition::inner_size

Product of dimensions after axis.

normalized_axis

int64_t Mila::Dnn::AxisPartition::normalized_axis

Axis normalized to [0, ndim).

num_slices

int64_t Mila::Dnn::AxisPartition::num_slices

outer_size * inner_size (for statistics buffers)

outer_size

int64_t Mila::Dnn::AxisPartition::outer_size

Product of dimensions before axis.

---

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

• /__w/Mila/Mila/Mila/Src/Dnn/Tensors/Tensor.Partitioning.ixx