Softmax. class torch.nn.Softmax(dim=None) [source]. 将Softmax函数应用于一个n维输入Tensor,对它们进行重新缩放,使n维输出Tensor的元素位于[0,1]的范围内, ...
確定! 回上一頁