# Tensors can be indexed just like Python lists. # Get the first 2 elements of each dimension (this is 4D tensor) A[[...]]