溫馨提示×

NumPy如何與PyTorch集成使用

小億
118
2024-05-13 13:15:14
欄目: 深度學習

要將NumPy和PyTorch集成使用,可以通過以下方式將NumPy數組轉換為PyTorch張量:

import numpy as np
import torch

# 創建一個NumPy數組
np_array = np.array([1, 2, 3, 4, 5])

# 將NumPy數組轉換為PyTorch張量
torch_tensor = torch.from_numpy(np_array)

# 打印PyTorch張量
print(torch_tensor)

反之,也可以將PyTorch張量轉換為NumPy數組:

import numpy as np
import torch

# 創建一個PyTorch張量
torch_tensor = torch.tensor([1, 2, 3, 4, 5])

# 將PyTorch張量轉換為NumPy數組
np_array = torch_tensor.numpy()

# 打印NumPy數組
print(np_array)

這樣,就可以在NumPy和PyTorch之間輕松地轉換數據,并實現兩者的集成使用。

0
亚洲午夜精品一区二区_中文无码日韩欧免_久久香蕉精品视频_欧美主播一区二区三区美女