206cfa4bb0fea0cee4c75223625c5e95b57d7010,python/tests/test_pytorch_transformer.py,,attn_values_sub_mask,#Any#Any#,81

Before Change

    B, H, T, _ = q.shape
    q = q.zero_()
    mask = subsequent_mask(T)
    res, _ = attn(q, k, v, mask=mask)
    res = res.numpy()
    gold = v.numpy()
    for b in range(B):
        for h in range(H):

After Change

    B, H, T, _ = q.shape
    q = q.zero_()
    mask = subsequent_mask(T)
    res = attn((q, k, v, mask))
    res = res.numpy()
    gold = v.numpy()
    for b in range(B):
        for h in range(H):

Instances