Hi Richard, thanks for your kind help. I tried and just got an error.
RuntimeError: gt() received an invalid combination of arguments - got (numpy.float32), but expected one of:

(float other)
didn’t match because some of the arguments have invalid types: (numpy.float32)
(Variable other)
didn’t match because some of the arguments have invalid types: (numpy.float32)