namespace Microsoft.ML.TorchSharp.AutoFormerV2
{
    public class BasicLayer : Module<Tensor, int, int, (Tensor, int, int, Tensor, int, int)>
    {
        public BasicLayer(int inChannels, int outChannels, int depth, int numHeads, int windowSize, double mlpRatio = 4, double dropRatio = 0, int localConvSize = 3, bool useShiftWindow = false, bool useInterpolate = false);
    }
}