How to add a new dimension to a PyTorch tensor? November 28, 2023 by Tarik a = torch.zeros(4, 5, 6) a = a[:, :, None, :] assert a.shape == (4, 5, 1, 6)