WrappedTensorGradientCallback#

class cuquantum.bindings.cudensitymat.WrappedTensorGradientCallback(
callback: TensorGradientCallbackType,
device: CallbackDevice,
direction: DifferentiationDir = DifferentiationDir.BACKWARD,
)[source]#

Wrapped tensor gradient callback Python extension type.

Methods

__init__(
callback: TensorGradientCallbackType,
device: CallbackDevice,
direction: DifferentiationDir = DifferentiationDir.BACKWARD,
)#

Create a wrapped tensor gradient callback.

Parameters:

  • callback – The tensor gradient callback function.

  • device – The device on which the callback is invoked.

  • direction – The direction of differentiation.

Attributes

callback#

object

Type:

callback