move vmap doc

This commit is contained in:
Erpim 2022-11-03 20:35:37 +08:00
parent 0d90568241
commit 1b7225fbf0
5 changed files with 20 additions and 4 deletions

View File

@ -465,7 +465,6 @@ Parameter操作函数
mindspore.ops.derivative
mindspore.ops.jet
mindspore.ops.vmap
调试函数
----------------

View File

@ -57,6 +57,14 @@ mindspore
mindspore.jvp
mindspore.vjp
自动向量化
----------------
.. mscnautosummary::
:toctree: mindspore
mindspore.vmap
运行环境
---------

View File

@ -1,7 +1,7 @@
mindspore.ops.vmap
mindspore.vmap
==================
.. py:function:: mindspore.ops.vmap(fn, in_axes=0, out_axes=0)
.. py:function:: mindspore.vmap(fn, in_axes=0, out_axes=0)
自动向量化Vectorizing Mapvmap是一种用于沿参数轴映射函数 `fn` 的高阶函数。

View File

@ -465,7 +465,6 @@ Differential Functions
mindspore.ops.derivative
mindspore.ops.jet
mindspore.ops.vmap
Debugging Functions
-------------------

View File

@ -168,6 +168,16 @@ Automatic Differentiation
mindspore.jvp
mindspore.vjp
Automatic Vectorization
---------------------------------
.. autosummary::
:toctree: mindspore
:nosignatures:
:template: classtemplate.rst
mindspore.vmap
Context
--------