Bien, puedo encontrar este artículo y probablemente encontré un problema que ya encontré. Hoy actualicé pytorch a 1.6.0 y descubrí que la división entre tensor e int no se puede usar directamente '/'. Obviamente 1.5.0 se puede utilizar -_-. La compatibilidad vecina de la antorcha merece la pena quejarse.
Para este problema, solo mire el documento oficial :
https://pytorch.org/docs/stable/generated/torch.div.html
O mira mi solución:
Para la división entre el tensor A y el entero n:
result = A / n # not supported in torch 1.6.0
# solution
result = torch.floor_divide(A, n)Este floor_divide es equivalente a '//' en python, es decir, el resultado es un número entero (el número después de que se elimina el punto decimal) .
Si no desea este tipo de división y desea obtener un valor preciso con un punto decimal, puede:
result = torch.true_divide(A, n)Seleccione las dos divisiones anteriores según la situación específica para resolver este problema.