Baremetal-NN
Baremetal-NN API documentation
Loading...
Searching...
No Matches

◆ NN_ones()

Tensor * NN_ones ( size_t  ndim,
const size_t *  shape,
DataType  dtype 
)

Returns a tensor filled with the scalar value 1.

Parameters
ndimnumber of dimensions
shapeshape of tensor
dtypedata type
Returns
Tensor