Что означает «[:,:,:, 0: 1]»?

Question

Я наткнулся на модель глубокого обучения, часть кода которой выглядит следующим образом. Что означают множественные двоеточия в [:,:,:, 0:1]?

def get_alpha_xij(input_tensor):
    alpha = Lambda(lambda x: x[:,:,:, 0:1])(input_tensor)
    x_i_j = Lambda(lambda x: x[:,:,:, 1:])(input_tensor)    
    return alpha, x_i_j

Answer 1

Это способ нарезать многомерный массив. Каждое двоеточие без чего-либо рядом с ним - кроме запятой - выбирает все элементы этого измерения. Чек https://numpy.org/devdocs/user/basics.indexing.html