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

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.