@miguelvr Isn’t this still useful for other layers than Linear though?  For example, the input tensor is of shape [sample, frame, image], like video, and you may want to apply a convnet module for each time frame.  Please kindly correct me if I get this wrong.