def forward(self, position_labels):
    return self.pos_embedding_matrix[position_labels.view((- 1)), :].view(position_labels.size(0), position_labels.size(1), (- 1))
def forward(self, position_labels):
    return self.pos_embedding_matrix[position_labels.view((- 1)), :].view(position_labels.size(0), position_labels.size(1), (- 1))