6 references to PaddingMaskKey
Microsoft.ML.TorchSharp (6)
NasBert\Modules\Layers\EncConvLayer.cs (2)
83
if (param.ContainsKey(
PaddingMaskKey
)) selfAttentionPaddingMask = (torch.Tensor)param[
PaddingMaskKey
];
NasBert\Modules\Layers\SelfAttentionLayer.cs (2)
81
if (!(param.ContainsKey(AttentionMaskKey) && param.ContainsKey(
PaddingMaskKey
))) return false;
84
selfAttentionPaddingMask = (torch.Tensor)param[
PaddingMaskKey
];
NasBert\Modules\TransformerCell.cs (2)
68
{Layer.
PaddingMaskKey
, selfAttentionPaddingMask},
133
{Layer.
PaddingMaskKey
, selfAttentionPaddingMask},