解决TypeError: Tensor is unhashable if Tensor equality is enabled. Instead, use tensor.experimental_re