jax.experimental.pallas.mosaic_gpu.layout_cast#

jax.experimental.pallas.mosaic_gpu.layout_cast(x, new_layout)[源代码]#

转换给定数组的布局。

参数:

  • x (Any)

  • new_layout (Layout | ParameterizedLayout)