El tipo de tensor pytorch en tipos de datos comunes en pitón

Tensor variable y se supone que un tipo de pytorch, como sigue:

y = torch.sum(m)
print(y)
print(type(y))
print(y.item())
print(type(y.item()))

Usando y.item () se puede convertir en flotador, la salida del programa es el siguiente:

tensor(452.4124, device='cuda:0', grad_fn=<SumBackward0>)
<class 'torch.Tensor'>
452.4123840332031
<class 'float'>

Las diferentes versiones de manera pytorch para resolver el problema no puede como yo esto es pytorch1.2

Publicado 36 artículos originales · ganado elogios 11 · vistas 6540

Supongo que te gusta

Origin blog.csdn.net/t20134297/article/details/103850377
Recomendado
Clasificación