Pytorch日积月累
蛊惑one
坚持输出!!!
展开
-
for循环遍历张量
今天才发现原来for循环就能简单的遍历一个张量。虽然简单,但是很实用。原创 2022-11-25 14:52:11 · 845 阅读 · 1 评论 -
Tensor的张量切片
tensor的切片问题!!!原创 2022-10-19 20:03:05 · 536 阅读 · 0 评论 -
Channel Shuffle类
ShuffleNet 中引入了 channel shuffle, 用来进行不同分组的特征之间的信息流动, 以提高性能。而Detectron2使用的pytorch版本一般较低,没有channel shuffle这个类,因此编写这个轮子用于通道洗牌。实现了与1.11.0官方库相同的结果。官方文档:ChannelShuffle — PyTorch 1.11.0 documentation...原创 2022-06-07 20:51:35 · 1340 阅读 · 0 评论 -
Pytorch笔记
主要用于记录一些用过但不经常使用的pytorch方法,按照首字母排序。原创 2022-06-07 17:03:28 · 569 阅读 · 0 评论