Fix set_tensor_attr

fchollet · fchollet · commit 084b7e113d0a · 2024-10-01T14:05:34.000-07:00
diff --git a/keras/src/backend/common/tensor_attributes.py b/keras/src/backend/common/tensor_attributes.py
@@ -5,7 +5,7 @@
 
 def set_tensor_attr(tensor, attr, value):
     try:
-        setattr(tensor, "_keras_mask", value)
+        setattr(tensor, attr, value)
     except AttributeError:
         if value is None:
             return
