PyTorch: Tensors and autograd PyTorch: Tensors and autograd PyTorch: Defining New autograd Functions PyTorch: Defining New autograd Functions