Problem: I would like to slice a tensor/numpy array A of shape (w, x, y, z) and result in a tensor B of shape (w, x, y, 1). I have a list L
A
B
L