1 write to softmax
Microsoft.ML.TorchSharp (1)
AutoFormerV2\Attention.cs (1)
93
this.
softmax
= nn.Softmax(dim: -1);
3 references to softmax
Microsoft.ML.TorchSharp (3)
AutoFormerV2\Attention.cs (3)
122
attn = this.
softmax
.forward(attn);
126
attn = this.
softmax
.forward(attn);
146
softmax
.Dispose();