jax.experimental.pallas.Slice# 类 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
