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