namespace Microsoft.ML.TorchSharp.AutoFormerV2
{
    public class AutoFormerV2 : Module<Tensor, (Tensor, Tensor, Tensor)>
    {
        public AutoFormerV2(int numClasses, List<int> embedChannels, List<int> depths, List<int> numHeads, Device device = null);
    }
}