jax.Array.flatten

Contents

jax.Array.flatten#

abstract Array.flatten(order='C')[source]#

Flatten array into a 1-dimensional shape.

Refer to jax.numpy.ravel() for the full documentation.

Parameters:
Return type:

Array