- Pytorch Tensor转numpy
RuntimeError: Can't call numpy() on Variable that requires grad. Use var.detach().numpy()
代码:myTensor.cuda().numpy()
原因:原tensor带有梯度,如不需要保持此信息,可用如下方式解决。
解决:myTensor.cuda().detach().numpy()
补充:如果不.cuda()
则会报此错误TypeError: can't convert CUDA tensor to numpy. Use Tensor.cpu() to copy the tensor to host memory first
Pytorch
©著作权归作者所有,转载或内容合作请联系作者
- 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
- 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
- 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
推荐阅读更多精彩内容
- 像认真记录生活一样记录Bug. 未经允许,不得转载,谢谢~~ 1. 从autograd.Variable中取Te...
- 几乎所有的深度学习框架背后的设计核心都是张量和计算图,PyTorch也不例外,本章我们将学习PyTorch中的张量...
- 今天青石的票圈出镜率最高的,莫过于张艺谋的新片终于定档了。 一张满溢着水墨风的海报一次次的出现在票圈里,也就是老谋...