nn_reshape function

Reshape

Reshape

Reshape a tensor to the given shape.

nn_reshape(shape)

Arguments

  • shape: (integer())

    The desired output shape.