📌  相关文章
📜  将张量转换为 numpy 数组 - Python 代码示例

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

代码示例3
import torch

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

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