Softmax over 3 Tensor

In a paper, I read that they had a (77512) tenor and after applying a softmax function they got (11512) tensor. I can not understand How to implement this in pytorch using softmax function

@Dina_Ahmed please provide the paper link and more details