1 write to DropoutLayer
Microsoft.ML.TorchSharp (1)
NasBert\Modules\MultiHeadAttention.cs (1)
102DropoutLayer = torch.nn.Dropout(_dropout);
2 references to DropoutLayer
Microsoft.ML.TorchSharp (2)
NasBert\Modules\MultiHeadAttention.cs (2)
284attentionWeights = DropoutLayer.forward(attentionWeights); 395DropoutLayer.Dispose();