Pythonのtorch.nn.SiLUの使用方法

PyTorchの`torch.nn.SiLU`を使用するには、`torch`と`torch.nn`モジュールをインポートします。`torch.nn.SiLU`はPyTorchの活性化関数で、Swish活性化関数とも呼ばれます。以下が`torch.nn.SiLU`を使用する簡単な例です。

“`python
import torch
import torch.nn as nn

# 入力テンソルを作成
x = torch.randn(1, 10)

# SiLU活性化関数を初期化
silu = nn.SiLU()

# 入力をSiLU活性化関数に通す
output = silu(x)

print(output)
“`

この例では、まず`torch`と`torch.nn`モジュールをインポートします。次に、サイズが`(1, 10)`のランダムな入力テンソル`x`を作成します。続いて、`nn.SiLU()`を使用してSiLU活性化関数のインスタンス`silu`を初期化します。最後に、入力テンソル`x`を`silu`活性化関数に通して結果を出力変数`output`に格納します。最後に、出力結果`output`を出力します。

bannerAds