Mila::Dnn::MultiAxisPartition Struct Reference

Multi-axis partition for normalization over trailing dimensions. More...

Collaboration diagram for Mila::Dnn::MultiAxisPartition:

Public Attributes
shape_t	normalized_shape
	Shape of normalized dimensions.
int64_t	normalized_size
	Product of the normalized dimensions.
shape_t	outer_shape
	Shape of outer dimensions.
int64_t	outer_size
	Product of dimensions before normalized axes.

Detailed Description

Multi-axis partition for normalization over trailing dimensions.

Member Data Documentation

normalized_shape

shape_t Mila::Dnn::MultiAxisPartition::normalized_shape

Shape of normalized dimensions.

normalized_size

int64_t Mila::Dnn::MultiAxisPartition::normalized_size

Product of the normalized dimensions.

outer_shape

shape_t Mila::Dnn::MultiAxisPartition::outer_shape

Shape of outer dimensions.

outer_size

int64_t Mila::Dnn::MultiAxisPartition::outer_size

Product of dimensions before normalized axes.

---

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

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