Solved by ptrblck in post #2 
              
              
                You could just index your output: 
batch_size = 10
output = torch.randn(batch_size, 7, 8)
output[:, 0, 0]

This will return output at pos [0, 0] for every batch sample.