torch矩阵基本操作
torch矩阵基本操作
合并
torch.stack — 增加纬度的合并

torch.cat — 不增加纬度合并

torch.unsqueeze(input, dim) — 只增加一个纬度
https://blog.youkuaiyun.com/xiaoxiao_ziteng/article/details/114540010?utm_medium=distribute.pc_aggpage_search_result.none-task-blog-2aggregatepagefirst_rank_ecpm_v1~rank_v31_ecpm-1-114540010-null-null.pc_agg_new_rank&utm_term=torch%E5%A6%82%E4%BD%95%E7%BB%99%E5%BC%A0%E9%87%8F%E5%A2%9E%E5%8A%A0%E7%BB%B4%E5%BA%A6&spm=1000.2123.3001.4430
复制
增维
torch.unsqueeze(input, dim) — 只增加一个纬度
增维的目的是什么呢?
答:为了广播运算

3016

被折叠的 条评论
为什么被折叠?



