nvidia.dali.plugin.pytorch.fn

The following table lists all operations available in nvidia.dali.plugin.pytorch.fn module:

Function

Device support

Short description

torch_python_function

CPU, GPU

Executes a function that is operating on Torch tensors.