From 0cd74ba41a4fe688edf315a24f1b9e3c01b5baac Mon Sep 17 00:00:00 2001
From: Dmovic <944388576@qq.com>
Date: Thu, 15 Aug 2024 15:55:58 +0800
Subject: [PATCH] Fix square docstr

---
 python/oneflow/framework/docstr/math_ops.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/python/oneflow/framework/docstr/math_ops.py b/python/oneflow/framework/docstr/math_ops.py
index b2cd6dc171f..2c40b58737c 100644
--- a/python/oneflow/framework/docstr/math_ops.py
+++ b/python/oneflow/framework/docstr/math_ops.py
@@ -1354,7 +1354,7 @@
     r"""Returns a new tensor with the square of the elements of :attr:`input`.
 
         .. math::
-            \text{out}_{i} = \sqrt{\text{input}_{i}}
+            \text{out}_{i} = \text{input}_{i} ^ {2}
 
         Args:
             input (Tensor): the input tensor.