pytorch中的Tensor和ndarray

霸气de小男生 提交于 2019-12-09 17:17:29

Tensor和ndarray是深度学习中经常遇到的两个概念: 针对于pytorch
(1)所在的位置: cpu gpu一般Tensor是可以在cpu中也可以在gpu中的
空间位置转换: 把Tensor从cpu中移动到gpu中: Tensor.cuda()
把Tensor从gpu中移动到cpu中: Tensor.cpu()
(2)tensor和ndarray之间的转换:
tensor 到 ndarray: Tensor.numpy()
ndarray到Tensor : torch.from_numpy(ndarray类型的数据)

易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!