2 references to TimeStepKey
Microsoft.ML.TorchSharp (2)
NasBert\Modules\Embedding\SinusoidalPositionalEmbedding.cs (2)
117
if (param.ContainsKey(
TimeStepKey
)) timeStep = (torch.Tensor)param[
TimeStepKey
];