21 lines
905 B
ReStructuredText
21 lines
905 B
ReStructuredText
mindspore.ops.derivative
|
||
========================
|
||
|
||
.. py:function:: mindspore.ops.derivative(fn, primals, order)
|
||
|
||
计算函数或网络输出对输入的高阶微分。给定待求导函数的原始输入和求导的阶数n,将返回函数输出对输入的第n阶导数。输入的初始1阶导数在内部默认设置为1,其他阶设置为0。
|
||
|
||
.. note::
|
||
- 若 `primals` 是int型的tensor,会被转化成float32格式进行计算。
|
||
|
||
参数:
|
||
- **fn** (Union[Cell, function]) - 待求导的函数或网络。
|
||
- **primals** (Union[Tensor, tuple[Tensor]]) - `fn` 的输入。
|
||
- **order** (int) - 求导的阶数。
|
||
|
||
返回:
|
||
tuple(out_primals, out_series)
|
||
|
||
- **out_primals** (Union[Tensor, list[Tensor]]) - `fn(primals)` 的结果。
|
||
- **out_series** (Union[Tensor, list[Tensor]]) - `fn` 输出对输入的第n阶导数。
|