We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent efd4b00 commit 0961be8Copy full SHA for 0961be8
graph_net/torch/utils.py
@@ -136,7 +136,6 @@ def format_data(data):
136
137
def process_tensor_info(tensor_info, name_prefix="example_input"):
138
tensor_type = tensor_info.get("type")
139
-
140
info = tensor_info.get("info", {})
141
dtype = info.get("dtype", "torch.float")
142
shape = info.get("shape", [])
0 commit comments