Loop over tensor of vectors (size [batch_size, 1, 28])

Question

I have a network where a vector of size [1,28] is generated for every input in a batch (of size 128). Now I want to access the values in these vectors for the whole batch, s