I’m not sure I understand the use case correctly.
To access values in a tensor, you can use indexing as:
x = torch.randn(32, 128, 4, 4)
print(x[0, 0])

This would print the “first” 4x4 tensor.
Could you give an example, how your output should look like, please?