is avaiable not just in the inner of the loop for the Cross entropy fct. This also remove duplicate code.
