From 888c09a3dbf8413c3aa76076e49cf52460334bbd Mon Sep 17 00:00:00 2001
From: Mauro Sciancalepore <40116397+masc-it@users.noreply.github.com>
Date: Mon, 1 Apr 2024 12:08:25 +0200
Subject: [PATCH] add identity op (#1976)

---
 candle-onnx/src/eval.rs | 5 +++++
 1 file changed, 5 insertions(+)

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:?}"),
         }
     }