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

Functions

template<typename TElementType , typename MR = Compute::CpuMemoryResource>
requires ValidTensorType<TElementType> && std::is_base_of_v<Compute::MemoryResource, MR>
void Mila::Dnn::random (Tensor< TElementType, MR > &tensor, TElementType min, TElementType max)
 Initializes a tensor with random values within a specified range.
 
template<typename TElementType , typename MR >
requires ValidTensorType<TElementType>&& std::is_base_of_v<Compute::MemoryResource, MR>
void Mila::Dnn::xavier (Tensor< TElementType, MR > &tensor, size_t input_size, size_t output_size)
 Initializes a tensor with Xavier/Glorot uniform initialization.
 

Files

file  /home/runner/work/Mila/Mila/Mila/Src/Dnn/Tensor/TensorHelpers.ixx
 Provides utility functions for tensor initialization and manipulation.