PyTorch中Tensor与NumPy的相互转换方法

PyTorch中Tensor与NumPy的相互转换

将Tensor转换为NumPy：

import torch

x = torch.ones(3, 3)
y = x.numpy()

将NumPy转换为Tensor：

import numpy as np

x = np.ones((3, 3))
y = torch.from_numpy(x)

注意：

• 将Tensor转换为NumPy时，共享内存。这意味着，如果更改其中一个，另一个也会被更改。
• Tensor和NumPy之间的转换仅限于CPU上。