jax.experimental.pallas.mosaic_gpu.layout_cast#

jax.experimental.pallas.mosaic_gpu.layout_cast(x, new_layout)[原始碼]#

轉換給定陣列的 layout。

參數:

  • x (Any)

  • new_layout (Layout | ParameterizedLayout)