rm unnecessary ndim guards (hurting backward compatibility)