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

◆ NN_tensor()

Tensor * NN_tensor ( size_t  ndim,
const size_t *  shape,
DataType  dtype,
void *  data 
)

Create a new tensor

Parameters
ndimnumber of dimensions
shapeshape of tensor
dtypedata type
datapointer to data, if NULL, the data will be allocated
Returns
Tensor