PyTorch equivalent of Numpy Round?
numpy.round() optionally accepts a specified number of digits to round to. However, torch.round does not, and while it seems like PyTorch will conform to NumPy eventually, what are people's current solutions?
I just want a function like torch.round(3.22, decimals=1) that returns 3.2.

You can define your own rounding function by
def round(x, decimals=0):
    b = 10**decimals
    return torch.round(x*b)/b

