pytorch中的Tensor和ndarray
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类型的数据) 来源: CSDN 作者: nbxuwentao 链接: https://blog.csdn.net/nbxuwentao/article/details/103460444