论文《Non-localNeuralNetworks》为了满足即插即用的功能，本博客重写nonlocal块，并可以根据自己的喜好选择2D卷积或者3D卷积，并可以选择是否使用bn层或pool。nonlocalblock模块图3D代码如下：classNonlocal_3d(nn.Module):def__init__(self,dim,dim_inner,pool