函数作用 计算两个tensor的矩阵乘法，torch.bmma,b),tensor a 的size为b,h,w),tensor b的size为b,w,m) 也就是说两个tensor的第一 … Continue reading torch.bmm函数解读
函数作用 计算两个tensor的矩阵乘法，torch.bmma,b),tensor a 的size为b,h,w),tensor b的size为b,w,m) 也就是说两个tensor的第一 … Continue reading torch.bmm函数解读