import TargetDrop
class MyNet():
    def __init__(self):
        ...
        self.target_drop=TargetDrop(in_channels)
    def forward(self, x):
        ...
        x=self.target_drop(x)