x_train[start_row:end_row, start_col:end_col]

例子:

x_train[::2, :784]#按步长取行,列取前784,如果代表所有列则可以省略,如下同样作用的代码
x_train[::2]