TF Mask in kaldi