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 ·
2828 阅读 ·
2 评论