提交 e64760fe 编写于 作者: M Megvii Engine Team

docs(mge): warp_perspective, gather & scatter

GitOrigin-RevId: f2436e850177b061c6bf7376d3f5b786baa6143c
上级 7d8a7e3e
...@@ -884,6 +884,10 @@ def warp_perspective( ...@@ -884,6 +884,10 @@ def warp_perspective(
:param interp_mode: interpolation methods. Default: "LINEAR" :param interp_mode: interpolation methods. Default: "LINEAR"
:return: output tensor. :return: output tensor.
Note:
The transformation matrix is the inverse of that used by `cv2.warpPerspective`.
Examples: Examples:
.. testcode:: .. testcode::
......
...@@ -391,6 +391,7 @@ def _get_idx(index, axis): ...@@ -391,6 +391,7 @@ def _get_idx(index, axis):
def gather(inp: Tensor, axis: int, index: Tensor) -> Tensor: def gather(inp: Tensor, axis: int, index: Tensor) -> Tensor:
# TODO: rewrite doc
r"""Gathers data from input tensor on axis using index. r"""Gathers data from input tensor on axis using index.
For a 3-D tensor, the output is specified by:: For a 3-D tensor, the output is specified by::
...@@ -462,6 +463,7 @@ def gather(inp: Tensor, axis: int, index: Tensor) -> Tensor: ...@@ -462,6 +463,7 @@ def gather(inp: Tensor, axis: int, index: Tensor) -> Tensor:
def scatter(inp: Tensor, axis: int, index: Tensor, source: Tensor) -> Tensor: def scatter(inp: Tensor, axis: int, index: Tensor, source: Tensor) -> Tensor:
# TODO: rewrite doc
r"""Writes all values from the tensor source into input tensor r"""Writes all values from the tensor source into input tensor
at the indices specified in the index tensor. at the indices specified in the index tensor.
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册