pytorch中维度不一致张量之间加减/逻辑运算问题
最近在看一个程序的时候很迷惑pytorch中维度不一致的张量之间如何让做加减和逻辑运算,参考了一个大神的博文https://blog.csdn.net/bxdzyhx/article/details/110831005
然后自己试了一下,特此记录一下
1.如果运算的两个张量维度不一致,那么首先把维度低的那个张量从右边和维度高的张量对齐
2.当两个张量维度相同时,对应轴的值需要一样,或者为1(即对应的每个轴要么相等,要么一方是1,才可以运算)
以逻辑与&为例
这里bp维度是(5,1,1,5),a的维度是(5,5)
进行逻辑与&运算: