jax.Array.flatten

abstract Array.flatten(order='C', *, out_sharding=None)

将数组展平为一维形状。

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

参数:

• self (Array)

• order (str)

返回类型:

Array