AttributeError: module 'torch' has no attribute 'swapaxes'

For some reason when using PyTorch on Google Colab, I'm unable to use torch.swapaxes and returns this error. Am I missing some additional import or something?