torch.nn.init.uniform_
function uniform_(tensor: Tensor, options?: InitUniformOptions): Tensorfunction uniform_(tensor: Tensor, a: number, b: number, options?: InitUniformOptions): TensorFill the input Tensor with values drawn from the uniform distribution U(a, b).
Parameters
tensorTensor- An n-dimensional Tensor
optionsInitUniformOptionsoptional- Optional settings for uniform initialization
Returns
Tensor– The input tensor filled with uniform valuesExamples
const w = torch.empty(3, 5);
torch.nn.init.uniform_(w, { a: -1, b: 1 });