#13685: Add note to the doc

tenstorrent · Nov 4, 2024 · 7caffc9 · 7caffc9
1 parent d46ba83
commit 7caffc9
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/ttnn/cpp/ttnn/operations/embedding_backward/embedding_backward_pybind.cpp b/ttnn/cpp/ttnn/operations/embedding_backward/embedding_backward_pybind.cpp
@@ -35,6 +35,8 @@ void py_bind_embedding_backward(py::module& module) {
         Returns:
             ttnn.Tensor: the output tensor.
 
+        Note:
+            The input and the output gradients must have the same datatype.
 
         Example:
             >>> device_id = 0