在Python中,您可以使用`import`關鍵字來導入`torch`模塊。`torch`是一個用于機器學習和深度學習的開源庫,因此需要先安裝`torch`庫。
首先,確保已經正確安裝了`torch`庫。您可以使用以下命令通過pip安裝:
pip install torch
一旦安裝完成,您可以在Python腳本中使用以下方式導入`torch`模塊:
import torch
然后就可以使用`torch`模塊中的各種功能了。例如,您可以創建張量、定義模型、進行訓練等。
import torch # 創建一個張量 x = torch.tensor([1, 2, 3]) # 定義一個簡單的神經網絡模型 model = torch.nn.Linear(10, 2) # 進行模型訓練 ...
請注意,如果您還希望使用其他`torch`模塊下的子模塊或類,可以使用點操作符進行導入。例如,如果要導入`torch.nn`模塊下的`Linear`類,可以使用以下方式:
import torch.nn as nn model = nn.Linear(10, 2)
這樣,您就可以根據需要導入和使用`torch`及其子模塊了。