Mila
Deep Neural Network Library
Loading...
Searching...
No Matches
Mila::Dnn::MultiAxisPartition Struct Referenceexport

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

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.


The documentation for this struct was generated from the following file:
  • Mila/Src/Dnn/Tensors/Tensor.Partitioning.ixx