If we use a onehot data type as the mask, its performance can be greatly improved.