📜  将 numpy 数组转换为张量 - Python 代码示例

📅  最后修改于: 2022-03-11 14:47:19.092000             🧑  作者: Mango

代码示例5
import torch

# Create PyTorch tensor
A_torch = torch.tensor([1, 2])

# Convert tensor to NumPy array
A_np = A_torch.numpy()