1 reference to Attention
Microsoft.ML.TorchSharp (1)
AutoFormerV2\AutoFormerV2Block.cs (1)
58
this.attn = new
Attention
(inChannels, headChannels, numHeads, attnRatio: 1, windowResolution: windowResolution);