4 references to AttentionMaskKey
Microsoft.ML.TorchSharp (4)
NasBert\Modules\Layers\SelfAttentionLayer.cs (2)
81
if (!(param.ContainsKey(
AttentionMaskKey
) && param.ContainsKey(PaddingMaskKey))) return false;
83
selfAttentionMask = (torch.Tensor)param[
AttentionMaskKey
];
NasBert\Modules\TransformerCell.cs (2)
67
{Layer.
AttentionMaskKey
, selfAttentionMask},
132
{Layer.
AttentionMaskKey
, selfAttentionMask},