3 overrides of CreateTargetsTensor
Microsoft.ML.TorchSharp (3)
NasBert\NerTrainer.cs (1)
136
private protected override torch.Tensor
CreateTargetsTensor
(ref List<TargetType> targets, torch.Device device)
NasBert\SentenceSimilarityTrainer.cs (1)
127
private protected override torch.Tensor
CreateTargetsTensor
(ref List<float> targets, torch.Device device)
NasBert\TextClassificationTrainer.cs (1)
128
private protected override torch.Tensor
CreateTargetsTensor
(ref List<long> targets, torch.Device device)
2 references to CreateTargetsTensor
Microsoft.ML.TorchSharp (2)
TorchSharpBaseTrainer.cs (2)
259
var targetsTensor =
CreateTargetsTensor
(ref targets, device: Device);
335
var targetsTensor =
CreateTargetsTensor
(ref targets, device: Device);