Diferencia del método pytorch squeeze () y unsqueeze ()

1 Aumente la dimensión sin apretar ()

tensor = tensor.unsqueeze(0)

2 Reducir la dimensión squeeze ()

tensor = tensor.squeeze(0)

Supongo que te gusta

Origin blog.csdn.net/ao1886/article/details/109152433
Recomendado
Clasificación