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