Update embedding_features.py

Qihoo360 · Jan 8, 2025 · a1191ba · a1191ba
1 parent 0e93816
commit a1191ba
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tensornet/layers/embedding_features.py b/tensornet/layers/embedding_features.py
@@ -287,7 +287,7 @@ def call(self, features, cols_to_output_tensors=None):
             if cols_to_output_tensors is not None:
                 cols_to_output_tensors[column] = new_tensor
 
-            output_tensors.append(masked_tensor)
+            output_tensors.append(new_tensor)
 
         if self.is_concat:
             return self._verify_and_concat_tensors(output_tensors)