fix doc of ms_function

This commit is contained in:
chenzhuo 2022-07-12 19:21:43 +08:00
parent 4be29b46b5
commit 743bbb4e18
1 changed files with 2 additions and 2 deletions

View File

@ -491,8 +491,8 @@ def ms_function(fn=None, input_signature=None, hash_args=None, jit_config=None):
... return fn(a)
... return inner_fn(x)
...
... inputs = Tensor(np.ones([10, 10, 10]).astype(np.float32))
... for i in range(10):
>>> inputs = Tensor(np.ones([10, 10, 10]).astype(np.float32))
>>> for i in range(10):
... closure_fn(inputs, func)
"""