本文介绍如何将NumPy数组转换为PyTorch所需的int64类型数据。通过对比astype()方法和np.int64()函数，指出后者能够正确转换数据类型。
Pytorch 需要 int 类型的数据，
1.astype
import numpy as np
arr = np.array([1,2,3,4,5]).astype(int)
print(arr.dtype)
## int32
输出的int32 并不是 pytorch 需要的类型
2.np.int64()
import numpy as np
arr = np.int64(np.array([1,2,3,4,5]))
print(arr.dtype)
## int64
输出的 int64 是pytorch需要的类型
扫一扫
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
