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

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.


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