jax.Array.reshape

抽象 Array.reshape(*args, order='C', out_sharding=None)

返回一个包含相同数据但具有新形状的数组。

请参阅 jax.numpy.reshape() 获取完整文档。

参数:

• self (Array)

• args (Any)

• order (str)

返回类型:

Array