pytorch转tensorrt_pytorch是啥

报错

原因分析

涉及到梯度传播的tensor变量不能够直接numpy()转numpy，需要使用：

代码语言：javascript复制

var.detach().numpy()

发布者：全栈程序员栈长，转载请注明出处：https://javaforall.cn/183816.html原文链接：https://javaforall.cn

numpy java https 网络安全

0 人点赞