I have video features (Batch, Time, Spatial, Feature_dim), like (32,15,4,2048).
and I need first a dimension transform using a nn.Liear(2048,512) .
But some features are masked according the associating mask (B,T,S) (True means yes,False means no)
How do I avoid updating this layer with unwanted features ?