jax.experimental.pallas.Slice

class jax.experimental.pallas.Slice(start, size, stride=1)

具有起始索引和大小的切片。

起始索引和大小可以是靜態的，即在追蹤和編譯時已知，也可以是動態的。

參數:

• start (int | Array)

• size (int | Array)

• stride (int)

__init__(start, size, stride=1)

參數:

• start (int | Array)

• size (int | Array)

• stride (int)

回傳型別:

None

方法

__init__(start, size[, stride])
from_slice(slc, size)
tree_flatten()
tree_unflatten(aux_data, children)

屬性

is_dynamic_size
is_dynamic_start
stride
start
size