torch.round函数

torch.round函数是PyTorch中的一个数学函数，用于将输入张量中的元素四舍五入到最近的整数。该函数的语法如下：

torch.round(input, out=None)

其中，参数input是需要进行四舍五入的张量，参数out是一个可选的输出张量，用于存储结果。如果没有指定参数out，则会创建一个新的张量来存储结果。

以下是一个示例：

import torch

x = torch.tensor([1.4, 2.6, 3.5, 4.2])
y = torch.round(x)
print(y)

输出结果为：

tensor([1., 3., 4., 4.])

在上面的示例中，我们创建了一个包含四个浮点数的张量x，然后使用torch.round函数将其四舍五入到最近的整数，得到了一个新的张量y。