2 references to PrepareBatchTensor
Microsoft.ML.TorchSharp (2)
Roberta\QATrainer.cs (2)
352
srcTensor =
PrepareBatchTensor
(ref inputTensors, device: Device, Tokenizer.RobertaModel().PadIndex);
353
targetTensor =
PrepareBatchTensor
(ref targetTensors, device: Device, 0);