工具

用于实用函数的内部模块。

函数

get_sliced_tensor	根据切片获取张量。
get_sliced_tensor_by_slices	根据活动切片获取张量。

get_sliced_tensor(mod, tensor, *hp_names)

根据切片获取张量。

Parameters:

• mod (动态模块) –

• tensor (Tensor | None) –

• hp_names (str | None) –

Return type:

张量 | 无

get_sliced_tensor_by_slices(tensor, slices)

根据活动切片获取张量。

Parameters:

• tensor (Tensor | None) –

• slices (List[slice | LongTensor]) –

Return type:

张量 | 无