We sometimes specify this by saying shape = (..., 3) and in the description you can mention that ... is the shape of the tensorfit.