../pruned_transducer_stateless2/joiner.py