Curious why this isn’t in PyTorch.