1 write to DropoutLayer
Microsoft.ML.TorchSharp (1)
NasBert\Modules\Layers\SelfAttentionLayer.cs (1)
39
DropoutLayer
= torch.nn.Dropout(dropoutRate);
2 references to DropoutLayer
Microsoft.ML.TorchSharp (2)
NasBert\Modules\Layers\SelfAttentionLayer.cs (2)
69
var dropout =
DropoutLayer
.forward(attention);
96
DropoutLayer
.Dispose();