- class torch.nn.Softmax(input,dim)或torch.nn.functional.softmax(input, dim)
作用:对n维输入张量运用Softmax函数,将张量的每个元素缩放到(0,1)区间且和为1。
参数:dim:指明维度,dim=0表示按列计算;dim=1表示按行计算。默认dim的方法已经弃用了,最好声明dim,否则会警告。
作用:对n维输入张量运用Softmax函数,将张量的每个元素缩放到(0,1)区间且和为1。
参数:dim:指明维度,dim=0表示按列计算;dim=1表示按行计算。默认dim的方法已经弃用了,最好声明dim,否则会警告。