|
Mila
Deep Neural Network Library
|
Classes | |
| struct | Mila::Dnn::TensorShape |
| Fixed-capacity inline shape descriptor for N-dimensional tensors. More... | |
Typedefs | |
| using | Mila::Dnn::dim_t = int64_t |
| Integer type used for tensor dimensions and indices. | |
| using | Mila::Dnn::index_t = TensorShape |
| Index descriptor for multi-dimensional element access. | |
| using | Mila::Dnn::shape_t = TensorShape |
| Row-major shape descriptor for tensor dimensional sizes. | |
| using | Mila::Dnn::stride_t = TensorShape |
| Stride descriptor (in elements) for each tensor dimension, row-major layout. | |
Functions | |
| constexpr dim_t | Mila::Dnn::elementCount (const shape_t &shape) noexcept |
| Product of a shape's extents – the number of logical elements it describes. | |
| std::string | Mila::Dnn::indexToString (const index_t &index) |
| int | Mila::Dnn::narrowToKernelIndex (dim_t value) |
| Narrow a dim_t to the 32-bit index type used inside device kernels. | |
| std::string | Mila::Dnn::shapeToString (const shape_t &shape) |
| std::string | Mila::Dnn::strideToString (const stride_t &stride) |
Files | |
| file | Mila/Src/Dnn/Tensors/Tensor.Types.ixx |
| Core shape, stride, and index types for the Mila tensor API. | |