pytorch实现softmax函数(tensor类型数据)
最直观的数学解释,softmax函数将输入的几个数据,变成0-1且之和为1的数据再输出。
公式如下:
意思是,每个经过缩放的元素经过exp函数后的结果,再除以结果之和。
举例:对一个二维tensor的列作为处理对象,将每一列的数据都处理成和为1的数据。
a = torch.rand(2,3)
print(a)
tensor([[0.4374, 0.0013, 0.7946],
[0.8905, 0.0954, 0.6671]])
这是创建了一个2x3的随机矩阵(0-1之间)
原创
2021-08-03 16:48:42 ·
2931 阅读 ·
2 评论