Added torch.round function