tensorflow predict example