时间:2023-05-04 02:44:19 阅读:144590 作者:1814
在Pytorch上遇到了can't convert CUDA tensor to numpy的问题
假设tensor为t
1、将tensor转换为cpu格式
t=t.CPU(2,进行numpy的对应操作
3、将数据转换为tensor,转换为gpu格式
t=Torch.from_numpy(t ).cuda ) )。
版权声明:该文观点仅代表作者本人。处理文章:请发送邮件至 三1五14八八95#扣扣.com 举报,一经查实,本站将立刻删除。