提交 b543e775 编写于 作者: R rogday

add test data for Div

上级 01e8a30a
......@@ -1558,3 +1558,29 @@ class SubFromConst1(nn.Module):
model = SubFromConst1()
input_ = Variable(torch.randn(1, 2, 3, 4, dtype=torch.float32))
save_data_and_model("sub_from_const1", input_, model)
########################## const / x ##########################
node = onnx.helper.make_node('Div', inputs=['x', 'y'], outputs=['z'])
x = np.array([2]).astype(np.float32)
y = np.array([[4, 4], [4, 4]]).astype(np.float32)
name = 'div_const'
input_files = os.path.join("data", "input_" + name)
np.save(input_files, x.data)
np.save(input_files, y.data)
z = (x / y).astype(np.float32)
output_files = os.path.join("data", "output_" + name)
np.save(output_files, np.ascontiguousarray(z.data))
X = onnx.helper.make_tensor('x', onnx.TensorProto.FLOAT, x.shape, x)
Y = onnx.helper.make_tensor_value_info('y', onnx.TensorProto.FLOAT, y.shape)
Z = onnx.helper.make_tensor_value_info('z', onnx.TensorProto.FLOAT, z.shape)
graph = onnx.helper.make_graph([node], 'div_const', [Y], [Z], initializer=[X])
model = onnx.helper.make_model(graph, producer_name=name)
models_files = os.path.join("models", name + ".onnx")
onnx.save(model, models_files)
########################## const / x ##########################
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册