jax.Array.reshape#

abstractArray.reshape(*args,order='C',out_sharding=None)[source]#

Returns an array containing the same data with a new shape.

Refer tojax.numpy.reshape() for full documentation.

Parameters:

• self (Array)

• args (Any)

• order (str)

Return type:

Array