|
Mila 0.13.48
Deep Neural Network Library
|
Classes | |
| class | Mila::Dnn::Tensor< TDataType, TMemoryResource > |
| Device-aware N-dimensional tensor. More... | |
Typedefs | |
| template<TensorDataType TDataType> | |
| using | Mila::Dnn::HostTensor = Tensor<TDataType, Compute::CpuMemoryResource> |
| Host tensor alias. | |
Functions | |
| template<TensorDataType TDataType, typename TMemoryResource> requires isValidTensor<TDataType, TMemoryResource> | |
| std::ostream & | Mila::Dnn::operator<< (std::ostream &os, const Tensor< TDataType, TMemoryResource > &tensor) |
| Stream insertion operator for tensor output. | |
Files | |
| file | /__w/Mila/Mila/Mila/Src/Dnn/Tensors/Tensor.ixx |
| Device-aware tensor type with scalar support. | |