Replace as_torch with torch.as_tensor
PyTorch has a function that does exactly what our as_torch is intended to do.
We can remove our redundant code and use the provided one.
PyTorch has a function that does exactly what our as_torch is intended to do.
We can remove our redundant code and use the provided one.