PyTorch inplace operator