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