Skip to content

Commit ff7e3f0

Browse files
committed
tensor name
1 parent 55277eb commit ff7e3f0

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

pyzoo/zoo/pipeline/api/net/tf_dataset.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,7 @@ def _to_tensor_structure(tensors):
4040
elif isinstance(tensors, dict):
4141
tensor_structure = {}
4242
for key, value in tensors.items():
43-
tensor_structure[key] = TensorMeta(dtype=value[0], shape=value[1], name=key)
43+
tensor_structure[key] = TensorMeta(dtype=value[0], shape=value[1], name="input_"+key)
4444
else:
4545
raise ValueError("In TFDataset.from_rdd, features and labels should be a tuple, "
4646
"a list of tuples or a dict of tuples")

0 commit comments

Comments
 (0)