Function neuronika::ones

pub fn ones<D: Dimension, Sh: ShapeBuilder<Dim = D>>(shape: Sh) -> Var<Input<D>>

Creates a variable with data filled with ones.

The shape is of type ndarray::ShapeBuilder.

Examples

use neuronika;
let t1 = neuronika::ones(1);
let t2 = neuronika::ones((1, 5));
let t3 = neuronika::ones([1, 2, 3]);

assert_eq!(t1.data().shape(), &[1]);
assert_eq!(t2.data().shape(), &[1, 5]);
assert_eq!(t3.data().shape(), &[1, 2, 3]);