Function tch::nn::init

pub fn init(i: Init, dims: &[i64], device: Device) -> Tensor

Creates a new float tensor with the specified shape, device, and initialization.