Base:
3 overrides of forward
Microsoft.ML.TorchSharp (3)
NasBert\Models\ModelPrediction.cs (1)
31
public override torch.Tensor
forward
(torch.Tensor srcTokens, torch.Tensor tokenMask = null)
NasBert\Models\NerModel.cs (1)
30
public override torch.Tensor
forward
(torch.Tensor srcTokens, torch.Tensor tokenMask = null)
Roberta\Models\RobertaModelForQA.cs (1)
33
public override torch.Tensor
forward
(torch.Tensor srcTokens, torch.Tensor tokenMask = null)
4 references to forward
Microsoft.ML.TorchSharp (4)
NasBert\NasBertTrainer.cs (4)
232
var logits = Model.
forward
(inputTensor);
283
logits = Model.
forward
(inputTensor, tokenMask: tokenMask);
290
logits = Model.
forward
(inputTensor);
608
return (Parent.Model as NasBertModel).
forward
(inputTensor);