Hi,
I wanted to ask that is it possible to convert an output of a NN to a long dtype in the custom loss function and still get a non-zero gradient?
As per my understanding, any of the operations like floor/ceil will give me a zero gradient.