fix: Out-Of-Bounds bug in Unsqueeze by gs-olive · Pull Request #1820 · pytorch/TensorRT (original) (raw)

torch.unsqueeze(torch.ones(4, 4), -1).shape

(4, 4, 1)

torch.unsqueeze(torch.ones(4, 4), -3).shape

(1, 4, 4)