对于一个矩阵,想要对行加和大于1的行使用softmax函数进行归一1.不要使用for循环一个个遍历,太慢啦。2.已经尝试过了pytorch中的torch.where函数,没有找到可以解决的方法,代码如下。3.最好使用pytorch或numpy或python自带的函数有的数据不符合预想的目标,但实在想不到合适的处理方式,谢谢大家帮助!
把数据处理成np,再手动求和一下,把每个item加一列数据和。再弄回去读这一列过softmax就行