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 33.

前のページ次のページ

[添付ファイル]