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