Tensor.ixx File Reference

Device-aware tensor type with scalar support. More...

#include <utility>
#include <vector>
#include <memory>
#include <string>
#include <cstdint>
#include <stdexcept>
#include <atomic>
#include <functional>
#include <numeric>
#include <iostream>
#include <sstream>
#include <iomanip>
#include <type_traits>
#include <limits>
#include <exception>
#include <optional>
#include <format>
#include <algorithm>
import Compute.DeviceRegistry;
import Compute.DeviceRegistrar;
import Dnn.TensorBuffer;
import Dnn.TensorHostTypeMap;
import Dnn.TensorDataTypeTraits;
import Dnn.ITensor;
import Compute.CpuDevice;
import Compute.MemoryResource;
import Dnn.TensorTypes;
import Dnn.TensorDataType;
import Dnn.TensorDataTypeMap;
import Compute.DeviceType;
import Compute.CpuTensorDataTypeTraits;
import Compute.CpuMemoryResource;
import Compute.Device;
import Compute.DeviceId;

Classes
class	Mila::Dnn::Tensor< TDataType, TMemoryResource >
	Device-aware N-dimensional tensor. More...
class	Mila::Dnn::UniqueIdGenerator
	Thread-safe generator for unique tensor identifiers. More...

Namespaces
namespace	Mila
	Mila main API namespace.
namespace	Mila::Dnn
namespace	Mila::Dnn::detail

Typedefs
template<TensorDataType TDataType>
using	Mila::Dnn::HostTensor = Tensor<TDataType, Compute::CpuMemoryResource>
	Host tensor alias.

Functions
std::string	Mila::Dnn::detail::formatBytes (size_t bytes)
template<TensorDataType TDataType>
constexpr size_t	Mila::Dnn::detail::getStorageSize (size_t logical_size)
	Calculates storage size in bytes for given logical element count.
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.

Detailed Description

Device-aware tensor type with scalar support.

Provides a device-bound N-dimensional tensor with explicit memory-resource abstraction and host/device access semantics. Scalars use empty shape {}.