From b42f1638351259675edc6aa234c1304a98e2c3ce Mon Sep 17 00:00:00 2001 From: "HE, Tao" Date: Fri, 12 Jan 2018 20:44:56 +0800 Subject: [ONNX] export sum, prod, sqrt improve log_softmax. (#4579) * ONNX: export sum, prod, sqrt improve log_softmax and fix a typo in doc. Signed-off-by: HE, Tao * Add new exported op to doc. Signed-off-by: HE, Tao * Double quotes. Signed-off-by: HE, Tao * Update trace log of log_softmax. Signed-off-by: HE, Tao * Improve export when dim is None and axes_i should be a list of ints. Signed-off-by: HE, Tao * Fix prod when no dim given. Signed-off-by: HE, Tao * Update line ends in test expected file. Signed-off-by: HE, Tao --- docs/source/onnx.rst | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) (limited to 'docs') diff --git a/docs/source/onnx.rst b/docs/source/onnx.rst index d1d8c9ffcf..fb8baf4357 100644 --- a/docs/source/onnx.rst +++ b/docs/source/onnx.rst @@ -130,9 +130,12 @@ The following operators are supported: * mm * addmm * neg +* sqrt * tanh * sigmoid * mean +* sum +* prod * t * expand (only when used before a broadcasting ONNX operator; e.g., add) * transpose @@ -189,7 +192,7 @@ for installing PyTorch from source. If the wanted operator is standardized in ONNX, it should be easy to add support for exporting such operator (adding a symbolic function for the operator). To confirm whether the operator is standardized or not, please check the -`ONNX operator list `_. +`ONNX operator list `_. If the operator is an ATen operator, which means you can find the declaration of the function in ``torch/csrc/autograd/generated/VariableType.h`` -- cgit v1.2.3