jax.Array.reshape# abstract Array.reshape(*args, order='C')[source]# Returns an array containing the same data with a new shape. Refer to jax.numpy.reshape() for full documentation. Parameters: self (Array) args (Any) order (str) Return type: Array
