Warning

This page was created from a pull request.

jax.lax.reshape

jax.lax.reshape(operand, new_sizes, dimensions=None)

Wraps XLA’s Reshape operator.

For inserting/removing dimensions of size 1, prefer using lax.squeeze / lax.expand_dims. These preserve information about axis identity that may be useful for advanced transformation rules.

Parameters

• operand (Any) –

• new_sizes (Sequence[int]) –

• dimensions (Optional[Sequence[int]]) –

Return type

Any