ExecutionCUDA

class nvmath.tensor.ExecutionCUDA(device_id: int = 0)

A data class for providing GPU execution options to the BinaryContraction and TernaryContraction objects, or the wrapper functions binary_contraction`and :func:`ternary_contraction().

device_id

CUDA device ordinal (only used if the operand resides on the CPU). The default value is 0.

Type:

int