Mila
Deep Neural Network Library
Loading...
Searching...
No Matches
Dnn.TensorTypes Module Reference

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.