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

docs(mge): warp_perspective, gather & scatter

GitOrigin-RevId: f2436e850177b061c6bf7376d3f5b786baa6143c
上级 7d8a7e3e
......@@ -884,6 +884,10 @@ def warp_perspective(
:param interp_mode: interpolation methods. Default: "LINEAR"
:return: output tensor.
Note:
The transformation matrix is the inverse of that used by `cv2.warpPerspective`.
Examples:
.. testcode::
......
......@@ -391,6 +391,7 @@ def _get_idx(index, axis):
def gather(inp: Tensor, axis: int, index: Tensor) -> Tensor:
# TODO: rewrite doc
r"""Gathers data from input tensor on axis using index.
For a 3-D tensor, the output is specified by::
......@@ -462,6 +463,7 @@ def gather(inp: Tensor, axis: int, index: 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
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.
先完成此消息的编辑!
想要评论请 注册