|
123456789101112131415161718192021222324252627282930313233 |
- mindspore.nn.Vjp
- =================
-
- .. py:class:: mindspore.nn.Vjp(fn)
-
- ſɱȻ(vector-Jacobian product, VJP)VJPӦģʽԶ֡
-
- ****
-
- **fn** (Cell) - Cell磬ڽ벢Ԫ顣
-
- **룺**
-
- - **inputs** (Tensor) - Σ
- - **v** (Tensor or Tuple of Tensor) - ſɱȾ˵״һ¡
-
- ****
-
- 2Ԫ鹹ɵԪ顣
-
- - **net_output** (Tensor or Tuple of Tensor) -
- - **vjp** (Tensor or Tuple of Tensor) - ſɱȻĽ
-
- ****
-
- >>> from mindspore.nn import Vjp
- >>> 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 = Vjp(Net())(x, y, v)
|