Note This tutorial will also work on AMD ROCm with no additional modifications. PyTorch offers a large library of operators that work on Tensors (e.g. torch.add, torch.sum, etc). However, you may wish ...