I\'d like to extract from a 5D tensor, [batch, H, W, F, C], some slices of it and do some computation. In particular, I\'m trying to extract for each position at a time of H and