244 views
2次元配列の列方向にsoftmaxの値を取得したい
#coding: UTF-8
import torch
import torch.nn as nn
device = torch.device("cpu")
data = [
[1.0, 0.0, 1.0],
[0.0, 0.0, 2.0],
[0.0, 0.0, 3.0]
]
#data = [10.0, 0.0, 0.0]
data = torch.tensor(data)
m = nn.Softmax(dim=0)
output = m(data)
print(output)
nn.Softmaxのdimを0にすると縦方向にsoftmaxの値を取得、dimの値を1にすると横方向にsoftmaxの値を取得する。
Page 19 of 29.
[添付ファイル]
お問い合わせ
マッスル
自己紹介
本サイトの作成者。
趣味：プログラム/水耕栽培/仮想通貨/激辛好き
プログラムは趣味と勉強を兼ねて、のんびり本サイトを作っています。
フレームワークはdjango。
仮想通貨はNEMが好き。
水耕栽培は激辛好きが高じて、キャロライナ・リーパーの栽培にチャレンジ中。
サイト/ブログ
https://www.osumoi-stdio.com/pyarticle/
