defined class for teacher forcing logits and updated the init function
defined class for teacher forcing logits and updated the init function

slundberg/shap,slundberg/shap,slundberg/shap,slundberg/shap