Supported Features

Each TorchAudio API supports a subset of PyTorch features, such as devices and data types. Supported features are indicated in API references like the following:

These icons mean that they are verified through automated testing.

Note

Missing feature icons mean that they are not tested, and this can mean different things, depending on the API.

1. The API is compatible with the feature but not tested.

2. The API is not compatible with the feature.

In case of 2, the API might explicitly raise an error, but that is not guaranteed. For example, APIs without an Autograd badge might throw an error during backpropagation, or silently return a wrong gradient.

If you use an API that hasn’t been labeled as supporting a feature, you might want to first verify that the feature works fine.

Devices

CPU

TorchAudio APIs that support CPU can perform their computation on CPU tensors.

CUDA

TorchAudio APIs that support CUDA can perform their computation on CUDA devices.

In case of functions, move the tensor arguments to CUDA device before passing them to a function.

For example:

cuda = torch.device("cuda")

waveform = waveform.to(cuda)
spectrogram = torchaudio.functional.spectrogram(waveform)

Classes with CUDA support are implemented with torch.nn.Module(). It is also necessary to move the instance to CUDA device, before passing CUDA tensors.

For example:

cuda = torch.device("cuda")

resampler = torchaudio.transforms.Resample(8000, 16000)
resampler.to(cuda)

waveform.to(cuda)
resampled = resampler(waveform)

Properties

Autograd

TorchAudio APIs with autograd support can correctly backpropagate gradients.

For the basics of autograd, please refer to this tutorial.

Note

APIs without this mark may or may not raise an error during backpropagation. The absence of an error raised during backpropagation does not necessarily mean the gradient is correct.

TorchScript

TorchAudio APIs with TorchScript support can be serialized and executed in non-Python environments.

For details on TorchScript, please refer to the documentation.