torch.has_autograd
Check if autograd is registered for an operation.
Parameters
opstring- The operation name
dtypeDTypeoptional- Optional dtype (checks dtype-specific autograd)
deviceDeviceoptional
Returns
boolean– True if autograd existsExamples
torch.library.has_autograd('mul'); // true
torch.library.has_autograd('custom_op', 'float32'); // checks dtype-specific