首页 > 编程知识 正文

tensorarray,tensor转array

时间:2023-05-04 01:46:12 阅读:144649 作者:2415

# model_out是CUDA上的tensor model _ out=model _ out.CPU (# detach ) :梯度model_out=model_out.detach ).nenach

版权声明:该文观点仅代表作者本人。处理文章:请发送邮件至 三1五14八八95#扣扣.com 举报,一经查实,本站将立刻删除。