commit
bd7d3e0765
|
@ -686,7 +686,7 @@ mindspore.Tensor
|
|||
|
||||
Tensor是否已经初始化。
|
||||
|
||||
.. py:method:: index_fill(dim, index, value):
|
||||
.. py:method:: index_fill(dim, index, value)
|
||||
|
||||
按 `index` 中给定的顺序选择索引,将输入 `value` 值填充到当前Tensor的所有 `dim` 维元素。
|
||||
|
||||
|
|
|
@ -12,7 +12,7 @@ mindspore.ms_function
|
|||
- **fn** (Function) - 要编译成图的Python函数。默认值:None。
|
||||
- **input_signature** (Tensor) - 用于表示输入参数的Tensor。Tensor的shape和dtype将作为函数的输入shape和dtype。默认值:None。
|
||||
- **hash_args** (Union[Object, List or Tuple of Objects]) - `fn` 里面用到的自由变量,比如外部函数或类对象,再次调用时若 `hash_args` 出现变化会触发重新编译。默认值:None。
|
||||
- **jit_config** (JitConfig) - 编译时所使用的JitConfig配置项,详细可参考 :class:`mindspore.JitConfig`。默认值:None。
|
||||
- **jit_config** (JitConfig) - 编译时所使用的JitConfig配置项,详细可参考 :class:`mindspore.JitConfig`。默认值:None。
|
||||
|
||||
.. note::
|
||||
- 如果指定了 `input_signature` ,则 `fn` 的每个输入都必须是Tensor。并且 `fn` 的输入参数将不会接受 `**kwargs` 参数。
|
||||
|
|
|
@ -1,5 +1,5 @@
|
|||
mindspore.ops.random_gamma
|
||||
====================
|
||||
==========================
|
||||
|
||||
.. py:function:: mindspore.ops.random_gamma(shape, alpha, seed=None)
|
||||
|
||||
|
|
|
@ -1,5 +1,5 @@
|
|||
mindspore.ops.unsorted_segment_sum
|
||||
================================
|
||||
==================================
|
||||
|
||||
.. py:function:: mindspore.ops.unsorted_segment_sum(input_x, segment_ids, num_segments)
|
||||
|
||||
|
|
Loading…
Reference in New Issue