5 references to ActivationFunction
Microsoft.ML.TorchSharp (5)
NasBert\Modules\Layers\EncConvLayer.cs (2)
30
("activation", new
ActivationFunction
(activationFn)),
37
("activation", new
ActivationFunction
(activationFn)),
NasBert\Modules\Layers\FeedForwardLayer.cs (1)
43
var activation = new
ActivationFunction
(activationFn);
NasBert\Modules\TransformerCell.cs (2)
51
_activationFn = new
ActivationFunction
(activationFn);
119
_activationFn = new
ActivationFunction
(activationFn);