Mila
Deep Neural Network Library
|
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. | |