diff --git a/testdata/dnn/onnx/data/input_exp.npy b/testdata/dnn/onnx/data/input_exp.npy new file mode 100644 index 000000000..09071c0f9 Binary files /dev/null and b/testdata/dnn/onnx/data/input_exp.npy differ diff --git a/testdata/dnn/onnx/data/output_exp.npy b/testdata/dnn/onnx/data/output_exp.npy new file mode 100644 index 000000000..d7e7b1bef Binary files /dev/null and b/testdata/dnn/onnx/data/output_exp.npy differ diff --git a/testdata/dnn/onnx/generate_onnx_models.py b/testdata/dnn/onnx/generate_onnx_models.py index f017ad775..83777c983 100644 --- a/testdata/dnn/onnx/generate_onnx_models.py +++ b/testdata/dnn/onnx/generate_onnx_models.py @@ -958,6 +958,14 @@ def forward(self, x): model = Power(2) save_data_and_model("pow2", x, model) +class Exp(nn.Module): + def forward(self, x): + return x.exp() + +x = Variable(torch.randn(2, 2)) +model = Exp() +save_data_and_model("exp", x, model) + class ReduceMaxGlobal(nn.Module): def forward(self, x): out = torch.max(x) diff --git a/testdata/dnn/onnx/models/exp.onnx b/testdata/dnn/onnx/models/exp.onnx new file mode 100644 index 000000000..2990fc312 --- /dev/null +++ b/testdata/dnn/onnx/models/exp.onnx @@ -0,0 +1,11 @@ +pytorch1.7:P + +01Exp_0"Exptorch-jit-exportZ +0 +  + +b +1 +  + +B \ No newline at end of file