pytorch中对于tensor的一些骚操作
使用环境:pytorch1.3.1;python3.6最近在读pytorch官方实现的Faster RCNN代码时,发现了一些非常规的tensor操作,为了方便理解,在此记录下:1. (a[:, None] * b[None, :]).view(-1)这是在计算anchor的k值出现的操作,其中a为一维向量[a1, a2, ..., an],b也为一维向量[b1, b2, ..., b...
原创
2019-12-12 20:56:41 ·
3136 阅读 ·
1 评论