m0_52126990的博客

原创 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之间）