não pode converter CUDA tensor para numpy. Use Tensor.cpu () para copiar o tensor a memória do host primeiro
NoSuchKey
Acho que você gosta
Origin blog.csdn.net/weixin_42528089/article/details/104926787
Recomendado
Clasificación