Función torch.round de PyTorch

Pytorch torch 参考手册Manual de referencia de PyTorch torch

torch.roundEs una función de PyTorch utilizada para redondear elemento por elemento.

Definición de la función

torch.round(input, decimals, out)

Ejemplo de uso

Ejemplo

import torch

x = torch.tensor([0.1, 0.5, 0.9, 1.2, 1.5])

print(torch.round(x))

# Especificar el número de decimales
y = torch.tensor([1.234, 5.678])
print(torch.round(y, decimals=2))

El resultado de salida es:

tensor([0., 0., 1., 1., 2.])
tensor([1.2300, 5.6800])

Pytorch torch 参考手册Manual de referencia de PyTorch torch

Otras extensiones