From 72ff7aeccb5875533f9ce87b1a231b8adb29ca11 Mon Sep 17 00:00:00 2001 From: Megvii Engine Team Date: Tue, 13 Jul 2021 23:56:29 +0800 Subject: [PATCH] feat(docs): add docs for megengine.functional.ones_like(jira #MGE-2702) GitOrigin-RevId: e808fc9e4e0f4614582e5767000545ea87738734 --- .../python/megengine/functional/tensor.py | 23 +++++++++++++++++++ 1 file changed, 23 insertions(+) diff --git a/imperative/python/megengine/functional/tensor.py b/imperative/python/megengine/functional/tensor.py index cab093a9..f87dbb81 100644 --- a/imperative/python/megengine/functional/tensor.py +++ b/imperative/python/megengine/functional/tensor.py @@ -183,6 +183,29 @@ def zeros_like(inp: Union[Tensor, SymbolVar]) -> Union[Tensor, SymbolVar]: def ones_like(inp: Union[Tensor, SymbolVar]) -> Union[Tensor, SymbolVar]: """ Returns a ones tensor with the same shape as input tensor. + + :param inp: input tensor. + :return: output ones tensor. + + Examples: + + .. testcode:: + + import numpy as np + from megengine import tensor + import megengine.functional as F + + inp = tensor(np.arange(1, 7, dtype=np.int32).reshape(2,3)) + out = F.ones_like(inp) + print(out.numpy()) + + Outputs: + + .. testoutput:: + + [[1 1 1] + [1 1 1]] + """ return full_like(inp, 1.0)