tensorflow transformer torch