Don’t I need to do the softmax again after applying the mask in order to get the new distribution of weights? Cause won’t doing batch matrix multiplication with negative infinity values really mess with it? Unless maybe the relu layer adjusts for that?