jax.lax.expand_dims

jax.lax.expand_dims(array, dimensions)

将任意数量的大小为1的维度插入数组。

参数:

• array (ArrayLike)

• dimensions (Sequence[int])

返回类型:

Array