目录
- 简介
- 场景描述
- 解决方法
- 举一反三
- 结语
简介
Hello！
非常感谢您阅读海轰的文章，倘若文中有错误的地方，欢迎您指出～
ଘ(੭ˊᵕˋ)੭
昵称：海轰
标签：程序猿｜C++选手｜学生
简介：因C语言结识编程，随后转入计算机专业，获得过国家奖学金，有幸在竞赛中拿过一些国奖、省奖…已保研
学习经验：扎实基础 + 多做笔记 + 多敲代码 + 多思考 + 学好英语！
唯有努力💪
本文仅记录自己感兴趣的内容
场景描述
已知一个二维数据，目标是将其转换为：每个元素转化为其占对应行元素之和的比例
例如：[[1,2],[2,3]] - > [[1/(1+2), 2/(1+2)], [2/(2+3), 3/(2+3)]]
解决方法
import torch
a = torch.tensor([[1,2],[2,3]])
a = torch.div(a, torch.stack(
[torch.sum(a, 1)] * a.size(1)).t() + 1e-10)
print(a)
举一反三
若是求每个元素转化为其占对应列元素之和的比例呢？？
import torch
a = torch.tensor([[1,2],[2,3]])
# 先转置一下
a = torch.transpose(a, 0 , 1)
a = torch.div(a, torch.stack(
[torch.sum(a, 1)] * a.size(1)).t() + 1e-10)
# 再转置回来
a = torch.transpose(a, 0 , 1)
print(a)
思路也比较简单：
- 求占行元素之和的比例最开始已经可以计算
- 求占列这时只需要对矩阵进行转置一次就好了（转为求占行），最后结果再转置回来即可
结语
文章仅作为个人学习笔记记录，记录从0到1的一个过程
希望对您有一点点帮助，如有错误欢迎小伙伴指正
