jax.lax.expand_dims#

jax.lax.expand_dims(array, dimensions)[source]#

Insert any number of size 1 dimensions into an array.

Parameters
Return type

Array