jax.experimental.pallas.mosaic_gpu.layout_cast# jax.experimental.pallas.mosaic_gpu.layout_cast(x, new_layout)[源代码]# 转换给定数组的布局。 参数: x (Any) new_layout (Layout)
