diff options
| -rw-r--r-- | candle-onnx/src/eval.rs | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/candle-onnx/src/eval.rs b/candle-onnx/src/eval.rs index cacb56ca..15cadf1d 100644 --- a/candle-onnx/src/eval.rs +++ b/candle-onnx/src/eval.rs @@ -776,6 +776,11 @@ pub fn simple_eval( let output = input.reshape(new_shape)?; values.insert(node.output[0].clone(), output); } + // https://github.com/onnx/onnx/blob/main/docs/Operators.md#identity + "Identity" => { + let input = get(&node.input[0])?; + values.insert(node.output[0].clone(), input.clone()); + } op_type => bail!("unsupported op_type {op_type} for op {node:?}"), } } |