cuquantum.OptimizerInfo

class cuquantum.OptimizerInfo(largest_intermediate: float, opt_cost: float, path: Iterable[Tuple[int, int]], slices: Iterable[Tuple[Hashable, int]])

A data class for capturing optimizer information.

largest_intermediate

The number of elements in the largest intermediate tensor. See CUTENSORNET_CONTRACTION_OPTIMIZER_INFO_LARGEST_TENSOR.

Type

float

opt_cost

The FLOP count of the optimized contraction path per slice. See CUTENSORNET_CONTRACTION_OPTIMIZER_INFO_FLOP_COUNT.

Type

float

path

The contraction path as a sequence of pairs in the numpy.einsum_path() format.

Type

Iterable[Tuple[int, int]]

slices

A sequence of (sliced mode, sliced extent) pairs.

Type

Iterable[Tuple[Hashable, int]]