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.])