kaldi-unsupervised/vimal: Added more testing functions to check NCE gradients