PythonでPyTorchを使い方を教えてください。

Pythonを使うためにPyTorchを使うには、まずPyTorchライブラリをインストールする必要があります。PyTorch公式ウェブサイトでインストールガイドを見つけ、あなたのオペレーティングシステムと設定に適したインストール方法を選択してください。

PyTorchが正しくインストールされ、実行されているかどうかを確認するために、以下のコードを使用できます:

import torch

# 创建一个张量
x = torch.tensor([[1, 2, 3], [4, 5, 6]])

# 打印张量
print(x)

# 运行在GPU上
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
x = x.to(device)

# 运行张量上的操作
y = torch.sum(x, dim=1)

# 打印结果
print(y)

これは、PyTorchを使用して、テンソルの作成、印刷、および操作を行う方法を示した簡単な例です。PyTorchの他の機能を使用して、より複雑なモデルや関数を作成し、深層学習の研究やアプリケーション開発を行うことができます。

bannerAds