3 references to PositionKey
Microsoft.ML.TorchSharp (3)
NasBert\Models\NasBertEncoder.cs (1)
207new Dictionary<string, object> { { PositionalEmbedding.PositionKey, positions } });
NasBert\Modules\Embedding\LearnedPositionalEmbedding.cs (2)
65if (param.ContainsKey(PositionKey)) positions = (torch.Tensor)param[PositionKey];