TensorFlow keeping shape the same when slicing?

Question

I am trying to take out a single element out of one dimension, while keeping the shapes the same.

The shape of the tensor is: (BATCH_SIZE, N_STEPS, NUM_FEATURES