import torch as th
points = th.ones(3, 4)
points_ndarray = points.numpy() # Tensor -> Ndarray
points _tensor = th.from_numpy(points_ndarray ) # Ndarray -> Tensor
Pytorch Tensor与Numpy ndarray的交互/相互转化
最新推荐文章于 2022-02-25 11:15:13 发布