When I want to change some specific values in the Variable, i run the code cross_entro[cross_entro==0.0] = 1.0. But I got the error info: *** AssertionError: MaskedFill can’t differentiate the mask .
Any ideas?