tch_plus::nn::init

Function f_init

Source 
pub fn f_init(
    i: Init,
    dims: &[i64],
    device: Device,
    kind: Kind,
) -> Result<Tensor, TchError>
Expand description

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