../pruned_transducer_stateless/joiner.py