|
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. | |