jax.Array.reshape

abstract Array.reshape(*args, order='C')

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

有关完整文档，请参阅 jax.numpy.reshape()。

参数:

• self (Array)

• args (Any)

• order (str)

返回类型:

Array