cuquantum.OptimizerInfo

class cuquantum.OptimizerInfo(largest_intermediate: float, opt_cost: float, path: Iterable[Tuple[int, int]], slices: Iterable[Tuple[Hashable, int]], num_slices: int, intermediate_modes: Iterable[Iterable[Hashable]])[源代码]

用于捕获优化器信息的数据类。

largest_intermediate

最大中间张量中的元素数量。参见 CUTENSORNET_CONTRACTION_OPTIMIZER_INFO_LARGEST_TENSOR。

类型

float

opt_cost

优化后的缩并路径的 FLOP 计数。参见 CUTENSORNET_CONTRACTION_OPTIMIZER_INFO_FLOP_COUNT。

类型

float

path

缩并路径，格式为 numpy.einsum_path() 中的成对序列。

类型

Iterable[Tuple[int, int]]

slices

(切片模式, 切片范围) 对的序列。

类型

Iterable[Tuple[Hashable, int]]

num_slices

切片的数量。

类型

int

intermediate_modes

所有中间张量的模式标签序列

类型

Iterable[Iterable[Hashable]]