PyTorchで何ができる？3分でわかる魅力

はじめに

PyTorchは「機械学習やりたい！」って人にピッタリの、使いやすくて超パワフルなライブラリです。今回はその魅力をサクッと紹介します！

1. ニューラルネットワークは自分で作る！

「自分でニューラルネット作りたい」って思ったら、PyTorchなら自由自在。例えばこんなシンプルなモデルも簡単。

import torch.nn as nn

class SimpleNN(nn.Module):
    def __init__(self):
        super(SimpleNN, self).__init__()
        self.fc = nn.Linear(10, 1)

    def forward(self, x):
        return self.fc(x)

複雑なモデルも簡単に拡張可能。カスタマイズしたい派にうってつけ！

2. 自動で微分計算（数学いらず！）

数式が苦手でも大丈夫！PyTorchのautogradが自動で微分してくれます。

import torch

x = torch.tensor(2.0, requires_grad=True)
y = x ** 2
y.backward()
print(x.grad)  # 4.0が出るよ！

数学の苦手なあなたも、これで勾配計算をマスター。

3. GPUで爆速学習

「計算が遅い…」と思ったら、GPUにお任せ！こんな風にGPUで処理できます。

device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
x = torch.randn(100, 100).to(device)

速さを体感したら、もうCPUには戻れないかも？

まとめ

PyTorchは柔軟でパワフル、そして簡単。自分だけのモデルを作り、数学に頼らず、GPUで高速計算！さあ、PyTorchを使って新しい世界に飛び込もう！