“ [：，：，::，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