|
Mila
Deep Neural Network Library
|
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. | |
Multi-axis partition for normalization over trailing dimensions.