mindspore2022/docs/api/api_python/nn/mindspore.nn.Jvp.rst

mindspore.nn.Jvp
=================

.. py:class:: mindspore.nn.Jvp(fn)

    �������������ſɱ�������(Jacobian-vector product, JVP)��JVP��Ӧǰ��ģʽ�Զ�΢�֡�

    **������**

    **fn** (Cell) - ����Cell�����磬���ڽ����������벢����������������Ԫ�顣

    **���룺**

    - **inputs** (Tensor) - ��������������������������
    - **v** (Tensor or Tuple of Tensor) - ���ſɱȾ����˵���������״�����������һ�¡�

    **�����**

    2������������Ԫ�鹹�ɵ�Ԫ�顣

    - **net_output** (Tensor or Tuple of Tensor) - ��������������������
    - **jvp** (Tensor or Tuple of Tensor) - �ſɱ��������Ľ����

    **������**

    >>> 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))