如何將列表或numpy數組轉換爲1d割炬張量？

Question

我有一個浮點值列表（或一個numpy數組）。我想創建一個包含所有這些值的割炬張量。我可以創建割炬張量並運行一個循環來存儲值。

但我想知道有沒有辦法，我可以用列表或數組中的初始值創建割炬張量？同時建議我是否有任何pythonic方式來實現這一點，因爲我在pytorch工作。

Answer 1

這些是pytorch中的一般操作，可在文檔中找到。 PyTorch允許與numpy輕鬆連接。有一種稱爲from_numpy的方法，並且文檔可用here

import numpy as np 
import torch 
array = np.arange(1, 11) 
tensor = torch.from_numpy(array)