Pytorchを理解したい。足し算をするだけのPytorchモデルをつくる。 - Qiita

機械学習モデルの多くはPytorchやTensorFlowで書かれています。 機械学習モデルに触れてみたいけど、Pytorchコードの部分がなんか「むにゃむにゃむにゃ。。。」という感じ。 シンプルなもので理解してみる 小学校の算数も、数を数えたり足し算をすることから始めた気がするので、 とりあえず、入力に１を足すモデルを作って理解してみます。 import torch.nn as nn device = "cuda" if torch.cuda.is_available() else "cpu" class AdditionModel(nn.Module): def __init__(self): super(AdditionModel, self).__init__() self.y = 1 def forward(self, x): x = torch.add(x,self.y) re