关于函数F.conv2d

[ZiMo-Chen]

F.conv2d(cur_point_mask, weight=torch.ones(1,1,background_length,background_length)　在这个函数为什么要设置它的权重为１，它的作用是什么

[XinyiYing]

F.conv2d(cur_point_mask, weight=torch.ones(1,1,background_length,background_length)的作用是将标签中的点，通过卷积的方式生成background_length * background_length的掩膜。该掩膜的生成是为了取出点标签的局部邻域。
当background_length=3的时候，点标签通过卷积生成了如下的3*3掩膜。
00000 00000
00000 01110
00100-01110
00000 01110
00000 00000