Information about axis partitioning of a tensor.
More...
|
|
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.
|
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