jax.lax.expand_dims# jax.lax.expand_dims(array, dimensions)[source]# 在数组中插入任意数量的大小为 1 的维度。 参数: array (ArrayLike) dimensions (Sequence[int]) 返回类型: Array
