I think my original question is perhaps a bit unclear, but what I want to avoid is having to explicitly mention the other dimensionalities, i.e. if there is an input which Im only certain of the last two dimensions I dont want to have to do a T.shape to get the others. But thanks for the suggestion.