Nigel
/
mindspore
forked from mindspore-Ecosystem/mindspore
0
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity
mindspore/docs/api/api_python/nn/mindspore.nn.Jvp.rst

37 lines
1.0 KiB
ReStructuredText
Raw Normal View History

modify api annotation
2021-11-26 17:39:06 +08:00
mindspore.nn.Jvp
=================
add docs of JVP and VJP
2021-11-25 20:13:52 +08:00
modify api annotation
2021-11-26 17:39:06 +08:00
.. py:class:: mindspore.nn.Jvp(fn)
add docs of JVP and VJP
2021-11-25 20:13:52 +08:00
modify api annotation
2021-11-26 17:39:06 +08:00
<20><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>ſɱ<C5BF><C9B1><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>(Jacobian-vector product, JVP)<29><>JVP<56><50>Ӧǰ<D3A6><C7B0>ģʽ<C4A3>Զ<EFBFBD>΢<EFBFBD>֡<EFBFBD>
add docs of JVP and VJP
2021-11-25 20:13:52 +08:00
modify api annotation
2021-11-26 17:39:06 +08:00
**<2A><><EFBFBD><EFBFBD><EFBFBD><EFBFBD>**
add docs of JVP and VJP
2021-11-25 20:13:52 +08:00
modify index and error format.
2021-12-04 20:36:47 +08:00
- **fn** (Cell) - <20><><EFBFBD><EFBFBD>Cell<6C><6C><EFBFBD><EFBFBD><EFBFBD>磬<EFBFBD><E7A3AC><EFBFBD>ڽ<EFBFBD><DABD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>벢<EFBFBD><EBB2A2><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>Ԫ<EFBFBD>顣
add docs of JVP and VJP
2021-11-25 20:13:52 +08:00
modify api annotation
2021-11-26 17:39:06 +08:00
**<2A><><EFBFBD>룺**
add docs of JVP and VJP
2021-11-25 20:13:52 +08:00
modify api annotation
2021-11-26 17:39:06 +08:00
- **inputs** (Tensor) - <20><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>Σ<EFBFBD><CEA3><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>
- **v** (Tensor or Tuple of Tensor) - <20><><EFBFBD>ſɱȾ<C9B1><C8BE><EFBFBD><EFBFBD><EFBFBD><EFBFBD>˵<EFBFBD><CBB5><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>״<EFBFBD><D7B4><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>һ<EFBFBD>¡<EFBFBD>
**<2A><><EFBFBD><EFBFBD><EFBFBD><EFBFBD>**
2<><32><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>Ԫ<EFBFBD>鹹<EFBFBD>ɵ<EFBFBD>Ԫ<EFBFBD>顣
- **net_output** (Tensor or Tuple of Tensor) - <20><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>
- **jvp** (Tensor or Tuple of Tensor) - <20>ſɱ<C5BF><C9B1><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>Ľ<EFBFBD><C4BD><EFBFBD><EFBFBD><EFBFBD>
modify index and error format.
2021-12-04 20:36:47 +08:00
**֧<><D6A7>ƽ̨<C6BD><CCA8>**
``Ascend`` ``GPU`` ``CPU``
modify api annotation
2021-11-26 17:39:06 +08:00
**<2A><><EFBFBD><EFBFBD><EFBFBD><EFBFBD>**
>>> from mindspore.nn import Jvp
>>> class Net(nn.Cell):
... def construct(self, x, y):
... return x**3 + y
>>> x = Tensor(np.array([[1, 2], [3, 4]]).astype(np.float32))
>>> y = Tensor(np.array([[1, 2], [3, 4]]).astype(np.float32))
>>> v = Tensor(np.array([[1, 1], [1, 1]]).astype(np.float32))
>>> output = Jvp(Net())(x, y, (v, v))