trick

w = np.array([~])

if dim of w is 4 and only use first dim, we can abbreviate like this 

w[1,...] # = w[1,:,:,:]