Pytorch 中torch.nn.Softmax的dim引數使用含義涉及到多維tensor時,對softmax的引數dim總是很迷,下面用一個例子說明 import torch.nn as nn m.
確定! 回上一頁