Softmax クラスのインスタンスを作成する際、引数dimで軸を指定すればよい。#やってみよう… ... input = torch.randn(2, 3) print(input). Copied!
確定! 回上一頁