始める無料で始める

線形レイヤーネットワーク

Neural networks には多くのレイヤーが含まれますが、その多くは線形レイヤーです。まずは単一の線形レイヤーを理解することで、複雑さを加える前に仕組みを把握できます。

入力テンソルに線形レイヤーを適用し、出力を確認しましょう。

この演習はコースの一部です

PyTorchで学ぶIntroduction to Deep Learning

コースを見る

演習の手順

  • 入力として 3 つの特徴量を受け取り、2 つの出力を返す Linear レイヤーを作成します。
  • input_tensor をその線形レイヤーに通します。

実践的なインタラクティブ演習

このサンプルコードを完成させて、この演習に挑戦してみましょう。

import torch
import torch.nn as nn

input_tensor = torch.tensor([[0.3471, 0.4547, -0.2356]])

# Create a Linear layer
linear_layer = nn.____(
                         in_features=____, 
                         out_features=____
                         )

# Pass input_tensor through the linear layer
output = ____(____)

print(output)
コードを編集して実行