Python PyTorch round用法及代码示例

本文简要介绍python语言中 torch.round 的用法。

用法:

torch.round(input, *, out=None) → Tensor

参数：

input(Tensor) -输入张量。

关键字参数：

out(Tensor,可选的) -输出张量。

返回一个新的张量，其中 input 的每个元素都舍入到最接近的整数。

例子：

>>> a = torch.randn(4)
>>> a
tensor([ 0.9920,  0.6077,  0.9734, -1.0362])
>>> torch.round(a)
tensor([ 1.,  1.,  1., -1.])