8 references to ToArray
Microsoft.ML.TorchSharp (8)
NasBert\NerTrainer.cs (2)
383
var prediction = argmax.
ToArray
<long>();
429
var prediction = argmax.
ToArray
<long>();
NasBert\TextClassificationTrainer.cs (1)
335
var values = (outputCacher as BertTensorCacher).Result.cpu().
ToArray
<float>();
Roberta\Models\RobertaModel.cs (1)
109
var srcTokenArray = srcTokens[i].
ToArray
<int>();
Utils\MetricUtils.cs (4)
32
var startScores = predictStartScores.
ToArray
<float>();
33
var endScores = predictEndScores.
ToArray
<float>();
34
var starts = predictStarts.
ToArray
<long>();
35
var ends = predictEnds.
ToArray
<long>();