pytorch tensor 保留小数点后两位

在 PyTorch 中，你可以使用 `.round()` 方法来保留小数点后两位。这个方法会对 tensor 中的每个元素进行四舍五入操作。

以下是一个示例代码：

import torch

# 创建一个包含浮点数的 tensor
tensor = torch.tensor([1.234567, 2.345678, 3.456789])

# 使用 round() 方法保留小数点后两位
rounded_tensor = tensor.round(decimals=2)

print(rounded_tensor)

输出将是：

tensor([1.2300, 2.3500, 3.4600])

这样，tensor 中的每个元素都被保留了小数点后两位。请注意，输出结果仍然是一个 tensor。如果你需要将其转换为 Python 的浮点数，可以使用 `.item()` 方法。例如：`rounded_tensor.item()`。