nn_ones2d_f16()

Tensor2D_F16 * nn_ones2d_f16

(

size_t

shape[2]

)

Creates a 2D tensor with type F16 and initializes it to 1.

nn_ones2d_f16

Parameters

shape

The shape of the tensor.

Returns

The created tensor.