Dnn.TensorOps Module Reference

Exported Modules
module	Dnn.TensorOps.Base
module	Compute.CpuTensorOps

Functions
template<TensorDataType TDataType, typename TMemoryResource> requires isValidTensor<TDataType, TMemoryResource>
void	Mila::Dnn::zero (Tensor< TDataType, TMemoryResource > &tensor, IExecutionContext *exec_context=nullptr)
	Zero a tensor using the fastest backend implementation.
template<TensorDataType TDataType, typename TMemoryResource> requires isValidTensor<TDataType, TMemoryResource>
void	Mila::Dnn::split (const Tensor< TDataType, TMemoryResource > &input, Tensor< TDataType, TMemoryResource > &output_a, Tensor< TDataType, TMemoryResource > &output_b, IExecutionContext *exec_context=nullptr)
template<TensorDataType TDataType, typename TMemoryResource> requires isValidTensor<TDataType, TMemoryResource>
void	Mila::Dnn::split (const Tensor< TDataType, TMemoryResource > &input, Tensor< TDataType, TMemoryResource > &output_a, Tensor< TDataType, TMemoryResource > &output_b, Tensor< TDataType, TMemoryResource > &output_c, IExecutionContext *exec_context=nullptr)
template<TensorDataType TDataType, typename TMemoryResource> requires isValidTensor<TDataType, TMemoryResource>
void	Mila::Dnn::fill_normal (Tensor< TDataType, TMemoryResource > &tensor, float mean, float stddev, IExecutionContext *exec_context=nullptr)
template<TensorDataType TDataType, typename TMemoryResource> requires isValidTensor<TDataType, TMemoryResource>
void	Mila::Dnn::fill_uniform (Tensor< TDataType, TMemoryResource > &tensor, host_value_t< TDataType > min_val, host_value_t< TDataType > max_val, IExecutionContext *exec_context=nullptr)
template<TensorDataType TDataType, typename TMemoryResource> requires isValidTensor<TDataType, TMemoryResource>
void	Mila::Dnn::add (const Tensor< TDataType, TMemoryResource > &a, const Tensor< TDataType, TMemoryResource > &b, Tensor< TDataType, TMemoryResource > &result, IExecutionContext *exec_context=nullptr)
	Element-wise addition with optional ExecutionContext (device-dispatched).
template<TensorDataType TDataType, typename TMemoryResource> requires isValidTensor<TDataType, TMemoryResource>
void	Mila::Dnn::subtract (const Tensor< TDataType, TMemoryResource > &a, const Tensor< TDataType, TMemoryResource > &b, Tensor< TDataType, TMemoryResource > &result, IExecutionContext *exec_context=nullptr)
	Element-wise subtraction with optional ExecutionContext (device-dispatched).
template<TensorDataType TDataType, typename TMemoryResource> requires isValidTensor<TDataType, TMemoryResource>
void	Mila::Dnn::multiply (const Tensor< TDataType, TMemoryResource > &a, const Tensor< TDataType, TMemoryResource > &b, Tensor< TDataType, TMemoryResource > &result, IExecutionContext *exec_context=nullptr)
	Element-wise multiplication with optional ExecutionContext (device-dispatched).
template<TensorDataType TDataType, typename TMemoryResource> requires isValidTensor<TDataType, TMemoryResource>
void	Mila::Dnn::divide (const Tensor< TDataType, TMemoryResource > &a, const Tensor< TDataType, TMemoryResource > &b, Tensor< TDataType, TMemoryResource > &result, IExecutionContext *exec_context=nullptr)
	Element-wise division with optional ExecutionContext (device-dispatched).
template<TensorDataType TDataType, typename TMemoryResource> requires isValidTensor<TDataType, TMemoryResource>
float	Mila::Dnn::sum (const Tensor< TDataType, TMemoryResource > &tensor, IExecutionContext *exec_context=nullptr)
	Sum reduction with optional ExecutionContext (device-dispatched).
template<TensorDataType TDataType, typename TMemoryResource> requires isValidTensor<TDataType, TMemoryResource>
Tensor< TDataType, TMemoryResource >	Mila::Dnn::operator+ (const Tensor< TDataType, TMemoryResource > &a, const Tensor< TDataType, TMemoryResource > &b)
	Element-wise addition operator (always synchronous).
template<TensorDataType TDataType, typename TMemoryResource> requires isValidTensor<TDataType, TMemoryResource>
Tensor< TDataType, TMemoryResource >	Mila::Dnn::operator- (const Tensor< TDataType, TMemoryResource > &a, const Tensor< TDataType, TMemoryResource > &b)
	Element-wise subtraction operator (always synchronous).
template<TensorDataType TDataType, typename TMemoryResource> requires isValidTensor<TDataType, TMemoryResource>
Tensor< TDataType, TMemoryResource >	Mila::Dnn::operator* (const Tensor< TDataType, TMemoryResource > &a, const Tensor< TDataType, TMemoryResource > &b)
	Element-wise multiplication operator (always synchronous).
template<TensorDataType TDataType, typename TMemoryResource> requires isValidTensor<TDataType, TMemoryResource>
Tensor< TDataType, TMemoryResource >	Mila::Dnn::operator/ (const Tensor< TDataType, TMemoryResource > &a, const Tensor< TDataType, TMemoryResource > &b)
	Element-wise division operator (always synchronous).
template<TensorDataType TDataType, typename TMemoryResource> requires isValidTensor<TDataType, TMemoryResource>
void	Mila::Dnn::fill (Tensor< TDataType, TMemoryResource > &tensor, std::span< const host_value_t< TDataType > > host_values, IExecutionContext *exec_context=nullptr)
	Copy host values into a tensor with device dispatch and optional ExecutionContext.
template<TensorDataType TDataType, typename TMemoryResource> requires isValidTensor<TDataType, TMemoryResource>
void	Mila::Dnn::fill (Tensor< TDataType, TMemoryResource > &tensor, host_value_t< TDataType > host_value, IExecutionContext *exec_context=nullptr)
	Fill a tensor with a scalar host value (device-dispatched) with optional ExecutionContext.
template<TensorDataType TSrcDataType, typename TSrcMemoryResource, TensorDataType TDstDataType, typename TDstMemoryResource> requires isValidTensor<TSrcDataType, TSrcMemoryResource> && isValidTensor<TDstDataType, TDstMemoryResource>
void	Mila::Dnn::copy (const Tensor< TSrcDataType, TSrcMemoryResource > &src, Tensor< TDstDataType, TDstMemoryResource > &dst, IExecutionContext *exec_context=nullptr)
	Copies tensor data from source to destination tensor with optional ExecutionContext.
template<TensorDataType TDstDataType, TensorDataType TSrcDataType, typename TSrcMemoryResource> requires isValidTensor<TSrcDataType, TSrcMemoryResource> && isValidTensor<TDstDataType, CpuMemoryResource>
Tensor< TDstDataType, CpuMemoryResource >	Mila::Dnn::toHost (const Tensor< TSrcDataType, TSrcMemoryResource > &src, IExecutionContext *exec_context=nullptr)
	Create a host (CPU) tensor from src and copy data into it.
template<TensorDataType TDstDataType, typename TDstMemoryResource> requires isValidTensor<TDstDataType, TDstMemoryResource>
void	Mila::Dnn::copyFromBlob (const Serialization::ITensorBlob &blob, Tensor< TDstDataType, TDstMemoryResource > &dst, IExecutionContext *exec_context=nullptr)
template<TensorDataType TSrcDataType, TensorDataType TDstDataType, typename TDstMemoryResource> requires isValidTensor<TDstDataType, TDstMemoryResource>
void	Mila::Dnn::copyFromBlobWithConversion (const Serialization::ITensorBlob &blob, Tensor< TDstDataType, TDstMemoryResource > &dst, IExecutionContext *exec_context=nullptr)
	Copy a serialized blob into a destination tensor, converting element types.

Files
file	/__w/Mila/Mila/Mila/Src/Dnn/Tensors/Operations/TensorOps.ixx
file	/__w/Mila/Mila/Mila/Src/Dnn/Tensors/Operations/TensorOps.Fill.ixx
	High-level initializer helpers (device-dispatching) for tensors.
file	/__w/Mila/Mila/Mila/Src/Dnn/Tensors/Operations/TensorOps.Math.ixx
	Device-dispatching math helpers for tensor arithmetic operations.
file	/__w/Mila/Mila/Mila/Src/Dnn/Tensors/Operations/TensorOps.Random.ixx
	Device-dispatching random initialization for tensors.
file	/__w/Mila/Mila/Mila/Src/Dnn/Tensors/Operations/TensorOps.Structural.ixx
	Device-dispatched structural operations for tensors.
file	/__w/Mila/Mila/Mila/Src/Dnn/Tensors/Operations/TensorOps.Transfer.ixx
	Tensor transfer utilities � copy/dispatch helpers for tensor data movement.
file	/__w/Mila/Mila/Mila/Src/Dnn/Tensors/Operations/TensorOps.Zero.ixx
	Device-dispatched fast zero operation for tensor buffers.