鹤城北斗

赞

2

|

搜索文章

赞

文章( 2 ) 沸点( 0 )

详解vmamba中的矩阵分块重排方案

本周在深入研究vmamba的代码时，发现PatchMerging2D类与PatchExpand2D类实现的很巧妙，其中还涉及到张量重排的技巧，这里做一记录。首先总结如下，给定输入

鹤城北斗
1年前
1.1k
2
评论

pytorch二分类与多分类代码实现

之前有师弟问我关于pytorch二分类和多分类训练测试代码的写法，这里就总结一下要点。下面为方便叙述，会将变量与形状同步给出，格式为 x(n,c,h,w)。首先捋一下过程，我们省略模型构建步骤，剩下

鹤城北斗
1年前
517
3
评论

个人成就

文章被点赞 7

文章被阅读 3,757

加入于

2022-04-11