bug in CRF method 'add_boundary_energy'?

[vandeb69]

Dear,
In the CRF class method add_boundary_energy, shouldn't

keras-contrib/keras_contrib/layers/crf.py

Line 403 in 3fc5ef7

end_mask = K.cast(K.greater(self.shift_left(mask), mask), K.floatx())

be replaced with: end_mask = K.cast(K.greater(mask, self.shift_left(mask)), K.floatx()) ?

If mask is [[0., 1., 1., 1., 1., 0.]], end_mask would be [[1., 0., 0., 0., 0., 0.]] in stead of the intended [[0., 0., 0., 0., 1., 0.]]

[FanYaning]

https://github.com/tensorflow/addons/blob/master/tensorflow_addons/layers/crf.py#L184