Skip to content

Commit cd2f0c0

Browse files
committed
fix make_tensor_proto didn't set dtype correctly.
1 parent 2e3bc06 commit cd2f0c0

1 file changed

Lines changed: 2 additions & 0 deletions

File tree

src/TensorFlowNET.Core/Tensors/tensor_util.cs

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -123,6 +123,8 @@ public static TensorProto make_tensor_proto(object values, TF_DataType dtype = T
123123
}
124124
else
125125
values = Convert.ChangeType(values, new_system_dtype);
126+
127+
dtype = values.GetDataType();
126128
}
127129

128130
shape = shape ?? values.GetShape();

0 commit comments

Comments
 (0)