6 ms·

How do you mask the output of the softmax layer? The link you referenced doesn't provide specific details.