class torch.nn.Softmax(input, dim)或torch.nn.functional.softmax(input, dim)对n维输入张量运用Softmax函数,将张量的每个元素缩放到(0,1)区间且和为1。
確定! 回上一頁