<gh_stars>1-10
from torch import nn 

class LSM(nn.Module):
    def __init__(self,):
        super().__init__()
    def forward(self,x):
        return x[:,:1,:,:]
from torch import nn 

class LSM(nn.Module):
    def __init__(self,):
        super().__init__()
    def forward(self,x):
        return x[:,:1,:,:]