明明两个大小相等的矩阵a和b,在使用a[b==1]时却报错,报错内容是第一维度大小不等
在我的代码中gt_dis和pred_mask的大小是相等的
却报如下的错
最后发现是因为pred_mask是numpy类型,而gt_dis是tensor类型导致的
把pred_mask改成tensor即可
明明两个大小相等的矩阵a和b,在使用a[b==1]时却报错,报错内容是第一维度大小不等
在我的代码中gt_dis和pred_mask的大小是相等的
却报如下的错
最后发现是因为pred_mask是numpy类型,而gt_dis是tensor类型导致的
把pred_mask改成tensor即可