833 lines
20 KiB
ReStructuredText
833 lines
20 KiB
ReStructuredText
mindspore.ops
|
||
=============================
|
||
|
||
mindspore.ops提供了大量的function接口，包括神经网络函数、数学运算函数、Tensor操作函数、Parameter操作函数、微分函数等。
|
||
|
||
动态shape的支持情况详见 `ops接口动态shape支持情况 <https://mindspore.cn/docs/zh-CN/master/api_python/dynamic_shape_func.html>`_ 。
|
||
|
||
模块导入方法如下：
|
||
|
||
.. code-block::
|
||
|
||
from mindspore import ops
|
||
|
||
MindSpore中 `mindspore.ops` 接口与上一版本相比，新增、删除和支持平台的变化信息请参考 `mindspore.ops API接口变更 <https://gitee.com/mindspore/docs/blob/master/resource/api_updates/func_api_updates_cn.md>`_ 。
|
||
|
||
Tensor操作函数
|
||
----------------
|
||
|
||
Tensor创建
|
||
^^^^^^^^^^^^^
|
||
|
||
.. mscnplatformautosummary::
|
||
:toctree: ops
|
||
:nosignatures:
|
||
:template: classtemplate.rst
|
||
|
||
mindspore.ops.arange
|
||
mindspore.ops.eps
|
||
mindspore.ops.eye
|
||
mindspore.ops.fill
|
||
mindspore.ops.full
|
||
mindspore.ops.full_like
|
||
mindspore.ops.heaviside
|
||
mindspore.ops.linspace
|
||
mindspore.ops.logspace
|
||
mindspore.ops.move_to
|
||
mindspore.ops.one_hot
|
||
mindspore.ops.ones
|
||
mindspore.ops.ones_like
|
||
mindspore.ops.range
|
||
mindspore.ops.zeros
|
||
mindspore.ops.zeros_like
|
||
|
||
随机生成函数
|
||
^^^^^^^^^^^^^^^^
|
||
|
||
.. mscnplatformautosummary::
|
||
:toctree: ops
|
||
:nosignatures:
|
||
:template: classtemplate.rst
|
||
|
||
mindspore.ops.bernoulli
|
||
mindspore.ops.gamma
|
||
mindspore.ops.laplace
|
||
mindspore.ops.multinomial
|
||
mindspore.ops.multinomial_with_replacement
|
||
mindspore.ops.rand
|
||
mindspore.ops.rand_like
|
||
mindspore.ops.randint
|
||
mindspore.ops.randint_like
|
||
mindspore.ops.randn
|
||
mindspore.ops.randn_like
|
||
mindspore.ops.random_gamma
|
||
mindspore.ops.random_poisson
|
||
mindspore.ops.randperm
|
||
mindspore.ops.standard_laplace
|
||
mindspore.ops.standard_normal
|
||
mindspore.ops.uniform
|
||
|
||
Array操作
|
||
^^^^^^^^^^^^^^^^
|
||
|
||
.. mscnplatformautosummary::
|
||
:toctree: ops
|
||
:nosignatures:
|
||
:template: classtemplate.rst
|
||
|
||
mindspore.ops.argwhere
|
||
mindspore.ops.batch_to_space_nd
|
||
mindspore.ops.bincount
|
||
mindspore.ops.block_diag
|
||
mindspore.ops.broadcast_to
|
||
mindspore.ops.cat
|
||
mindspore.ops.channel_shuffle
|
||
mindspore.ops.chunk
|
||
mindspore.ops.column_stack
|
||
mindspore.ops.concat
|
||
mindspore.ops.conj
|
||
mindspore.ops.count_nonzero
|
||
mindspore.ops.deepcopy
|
||
mindspore.ops.diag
|
||
mindspore.ops.diagflat
|
||
mindspore.ops.diagonal
|
||
mindspore.ops.diagonal_scatter
|
||
mindspore.ops.diag_embed
|
||
mindspore.ops.dyn_shape
|
||
mindspore.ops.dsplit
|
||
mindspore.ops.dstack
|
||
mindspore.ops.einsum
|
||
mindspore.ops.expand_dims
|
||
mindspore.ops.flip
|
||
mindspore.ops.fliplr
|
||
mindspore.ops.flipud
|
||
mindspore.ops.gather
|
||
mindspore.ops.gather_d
|
||
mindspore.ops.gather_elements
|
||
mindspore.ops.gather_nd
|
||
mindspore.ops.hsplit
|
||
mindspore.ops.hstack
|
||
mindspore.ops.index_add
|
||
mindspore.ops.index_fill
|
||
mindspore.ops.index_select
|
||
mindspore.ops.inplace_add
|
||
mindspore.ops.inplace_index_add
|
||
mindspore.ops.inplace_sub
|
||
mindspore.ops.inplace_update
|
||
mindspore.ops.is_nonzero
|
||
mindspore.ops.masked_fill
|
||
mindspore.ops.masked_select
|
||
mindspore.ops.meshgrid
|
||
mindspore.ops.moveaxis
|
||
mindspore.ops.movedim
|
||
mindspore.ops.narrow
|
||
mindspore.ops.nan_to_num
|
||
mindspore.ops.nanmean
|
||
mindspore.ops.nanmedian
|
||
mindspore.ops.nansum
|
||
mindspore.ops.normal
|
||
mindspore.ops.nonzero
|
||
mindspore.ops.nsa_compress
|
||
mindspore.ops.nsa_compress_attention
|
||
mindspore.ops.numel
|
||
mindspore.ops.permute
|
||
mindspore.ops.population_count
|
||
mindspore.ops.rank
|
||
mindspore.ops.repeat_elements
|
||
mindspore.ops.repeat_interleave
|
||
mindspore.ops.reshape
|
||
mindspore.ops.reverse_sequence
|
||
mindspore.ops.roll
|
||
mindspore.ops.row_stack
|
||
mindspore.ops.scatter
|
||
mindspore.ops.scatter_nd
|
||
mindspore.ops.select
|
||
mindspore.ops.select_scatter
|
||
mindspore.ops.sequence_mask
|
||
mindspore.ops.shape
|
||
mindspore.ops.shuffle
|
||
mindspore.ops.size
|
||
mindspore.ops.slice
|
||
mindspore.ops.slice_scatter
|
||
mindspore.ops.sort
|
||
mindspore.ops.space_to_batch_nd
|
||
mindspore.ops.sparse_segment_mean
|
||
mindspore.ops.split
|
||
mindspore.ops.squeeze
|
||
mindspore.ops.stack
|
||
mindspore.ops.strided_slice
|
||
mindspore.ops.sum
|
||
mindspore.ops.swapaxes
|
||
mindspore.ops.swapdims
|
||
mindspore.ops.tensor_scatter_add
|
||
mindspore.ops.tensor_scatter_div
|
||
mindspore.ops.tensor_scatter_max
|
||
mindspore.ops.tensor_scatter_min
|
||
mindspore.ops.tensor_scatter_mul
|
||
mindspore.ops.tensor_scatter_sub
|
||
mindspore.ops.tensor_scatter_elements
|
||
mindspore.ops.tensor_split
|
||
mindspore.ops.tile
|
||
mindspore.ops.tril
|
||
mindspore.ops.triu
|
||
mindspore.ops.transpose
|
||
mindspore.ops.unbind
|
||
mindspore.ops.unique
|
||
mindspore.ops.unique_consecutive
|
||
mindspore.ops.unsorted_segment_max
|
||
mindspore.ops.unsorted_segment_min
|
||
mindspore.ops.unsorted_segment_prod
|
||
mindspore.ops.unsorted_segment_sum
|
||
mindspore.ops.unsqueeze
|
||
mindspore.ops.unstack
|
||
mindspore.ops.view_as_real
|
||
mindspore.ops.vsplit
|
||
mindspore.ops.vstack
|
||
mindspore.ops.where
|
||
mindspore.ops.cross
|
||
mindspore.ops.renorm
|
||
|
||
类型转换
|
||
^^^^^^^^^^^^^^^^
|
||
|
||
.. mscnplatformautosummary::
|
||
:toctree: ops
|
||
:nosignatures:
|
||
:template: classtemplate.rst
|
||
|
||
mindspore.ops.cast
|
||
mindspore.ops.is_tensor
|
||
mindspore.ops.scalar_to_tensor
|
||
mindspore.ops.tuple_to_array
|
||
|
||
梯度剪裁
|
||
^^^^^^^^
|
||
|
||
.. mscnplatformautosummary::
|
||
:toctree: ops
|
||
:nosignatures:
|
||
:template: classtemplate.rst
|
||
|
||
mindspore.ops.clip_by_global_norm
|
||
mindspore.ops.clip_by_value
|
||
mindspore.ops.clip_by_norm
|
||
|
||
数学运算函数
|
||
----------------
|
||
|
||
逐元素运算
|
||
^^^^^^^^^^^^^
|
||
|
||
.. mscnplatformautosummary::
|
||
:toctree: ops
|
||
:nosignatures:
|
||
:template: classtemplate.rst
|
||
|
||
mindspore.ops.abs
|
||
mindspore.ops.absolute
|
||
mindspore.ops.accumulate_n
|
||
mindspore.ops.acos
|
||
mindspore.ops.arccos
|
||
mindspore.ops.acosh
|
||
mindspore.ops.add
|
||
mindspore.ops.addcdiv
|
||
mindspore.ops.addcmul
|
||
mindspore.ops.addmv
|
||
mindspore.ops.addn
|
||
mindspore.ops.angle
|
||
mindspore.ops.arccosh
|
||
mindspore.ops.arcsin
|
||
mindspore.ops.arcsinh
|
||
mindspore.ops.arctan
|
||
mindspore.ops.arctanh
|
||
mindspore.ops.arctan2
|
||
mindspore.ops.asin
|
||
mindspore.ops.asinh
|
||
mindspore.ops.atan
|
||
mindspore.ops.atan2
|
||
mindspore.ops.atanh
|
||
mindspore.ops.atleast_1d
|
||
mindspore.ops.atleast_2d
|
||
mindspore.ops.atleast_3d
|
||
mindspore.ops.bessel_i0
|
||
mindspore.ops.bessel_i0e
|
||
mindspore.ops.bessel_i1
|
||
mindspore.ops.bessel_i1e
|
||
mindspore.ops.bessel_j0
|
||
mindspore.ops.bessel_j1
|
||
mindspore.ops.bessel_k0
|
||
mindspore.ops.bessel_k0e
|
||
mindspore.ops.bessel_k1
|
||
mindspore.ops.bessel_k1e
|
||
mindspore.ops.bessel_y0
|
||
mindspore.ops.bessel_y1
|
||
mindspore.ops.bitwise_and
|
||
mindspore.ops.bitwise_left_shift
|
||
mindspore.ops.bitwise_or
|
||
mindspore.ops.bitwise_right_shift
|
||
mindspore.ops.bitwise_xor
|
||
mindspore.ops.ceil
|
||
mindspore.ops.clamp
|
||
mindspore.ops.clip
|
||
mindspore.ops.combinations
|
||
mindspore.ops.copysign
|
||
mindspore.ops.cos
|
||
mindspore.ops.cosh
|
||
mindspore.ops.cosine_similarity
|
||
mindspore.ops.cov
|
||
mindspore.ops.diff
|
||
mindspore.ops.deg2rad
|
||
mindspore.ops.digamma
|
||
mindspore.ops.div
|
||
mindspore.ops.divide
|
||
mindspore.ops.erf
|
||
mindspore.ops.erfc
|
||
mindspore.ops.erfinv
|
||
mindspore.ops.exp
|
||
mindspore.ops.exp2
|
||
mindspore.ops.expm1
|
||
mindspore.ops.floor
|
||
mindspore.ops.floor_div
|
||
mindspore.ops.floor_divide
|
||
mindspore.ops.floor_mod
|
||
mindspore.ops.float_power
|
||
mindspore.ops.fmax
|
||
mindspore.ops.fmod
|
||
mindspore.ops.frac
|
||
mindspore.ops.gcd
|
||
mindspore.ops.hypot
|
||
mindspore.ops.igamma
|
||
mindspore.ops.igammac
|
||
mindspore.ops.imag
|
||
mindspore.ops.i0
|
||
mindspore.ops.inv
|
||
mindspore.ops.invert
|
||
mindspore.ops.lcm
|
||
mindspore.ops.ldexp
|
||
mindspore.ops.lerp
|
||
mindspore.ops.log
|
||
mindspore.ops.log2
|
||
mindspore.ops.log10
|
||
mindspore.ops.log1p
|
||
mindspore.ops.logaddexp
|
||
mindspore.ops.logaddexp2
|
||
mindspore.ops.logical_and
|
||
mindspore.ops.logical_not
|
||
mindspore.ops.logical_or
|
||
mindspore.ops.logical_xor
|
||
mindspore.ops.logit
|
||
mindspore.ops.mul
|
||
mindspore.ops.multiply
|
||
mindspore.ops.mvlgamma
|
||
mindspore.ops.neg
|
||
mindspore.ops.negative
|
||
mindspore.ops.nextafter
|
||
mindspore.ops.polar
|
||
mindspore.ops.polygamma
|
||
mindspore.ops.positive
|
||
mindspore.ops.pow
|
||
mindspore.ops.rad2deg
|
||
mindspore.ops.ravel
|
||
mindspore.ops.real
|
||
mindspore.ops.reciprocal
|
||
mindspore.ops.remainder
|
||
mindspore.ops.rot90
|
||
mindspore.ops.round
|
||
mindspore.ops.rsqrt
|
||
mindspore.ops.sgn
|
||
mindspore.ops.sign
|
||
mindspore.ops.signbit
|
||
mindspore.ops.sin
|
||
mindspore.ops.sinc
|
||
mindspore.ops.sinh
|
||
mindspore.ops.sqrt
|
||
mindspore.ops.square
|
||
mindspore.ops.sub
|
||
mindspore.ops.subtract
|
||
mindspore.ops.t
|
||
mindspore.ops.tan
|
||
mindspore.ops.tanhshrink
|
||
mindspore.ops.trapz
|
||
mindspore.ops.tril_indices
|
||
mindspore.ops.triu_indices
|
||
mindspore.ops.true_divide
|
||
mindspore.ops.trunc
|
||
mindspore.ops.truncate_div
|
||
mindspore.ops.truncate_mod
|
||
mindspore.ops.xdivy
|
||
mindspore.ops.xlogy
|
||
mindspore.ops.zeta
|
||
|
||
Reduction函数
|
||
^^^^^^^^^^^^^
|
||
|
||
.. mscnplatformautosummary::
|
||
:toctree: ops
|
||
:nosignatures:
|
||
:template: classtemplate.rst
|
||
|
||
mindspore.ops.all
|
||
mindspore.ops.amax
|
||
mindspore.ops.amin
|
||
mindspore.ops.aminmax
|
||
mindspore.ops.any
|
||
mindspore.ops.argmax
|
||
mindspore.ops.argmin
|
||
mindspore.ops.cummax
|
||
mindspore.ops.cummin
|
||
mindspore.ops.cumprod
|
||
mindspore.ops.cumsum
|
||
mindspore.ops.histc
|
||
mindspore.ops.logcumsumexp
|
||
mindspore.ops.logsumexp
|
||
mindspore.ops.max
|
||
mindspore.ops.mean
|
||
mindspore.ops.median
|
||
mindspore.ops.min
|
||
mindspore.ops.norm
|
||
mindspore.ops.prod
|
||
mindspore.ops.std
|
||
mindspore.ops.std_mean
|
||
mindspore.ops.var
|
||
mindspore.ops.var_mean
|
||
|
||
比较函数
|
||
^^^^^^^^^^^^^
|
||
|
||
.. mscnplatformautosummary::
|
||
:toctree: ops
|
||
:nosignatures:
|
||
:template: classtemplate.rst
|
||
|
||
mindspore.ops.argsort
|
||
mindspore.ops.approximate_equal
|
||
mindspore.ops.bucketize
|
||
mindspore.ops.eq
|
||
mindspore.ops.equal
|
||
mindspore.ops.ge
|
||
mindspore.ops.greater
|
||
mindspore.ops.greater_equal
|
||
mindspore.ops.gt
|
||
mindspore.ops.intopk
|
||
mindspore.ops.isclose
|
||
mindspore.ops.isfinite
|
||
mindspore.ops.isinf
|
||
mindspore.ops.isnan
|
||
mindspore.ops.isneginf
|
||
mindspore.ops.isposinf
|
||
mindspore.ops.isreal
|
||
mindspore.ops.is_complex
|
||
mindspore.ops.is_floating_point
|
||
mindspore.ops.le
|
||
mindspore.ops.less
|
||
mindspore.ops.less_equal
|
||
mindspore.ops.lt
|
||
mindspore.ops.maximum
|
||
mindspore.ops.minimum
|
||
mindspore.ops.msort
|
||
mindspore.ops.ne
|
||
mindspore.ops.not_equal
|
||
mindspore.ops.searchsorted
|
||
mindspore.ops.topk
|
||
|
||
线性代数函数
|
||
^^^^^^^^^^^^^
|
||
|
||
.. mscnplatformautosummary::
|
||
:toctree: ops
|
||
:nosignatures:
|
||
:template: classtemplate.rst
|
||
|
||
mindspore.ops.addbmm
|
||
mindspore.ops.addmm
|
||
mindspore.ops.addr
|
||
mindspore.ops.adjoint
|
||
mindspore.ops.baddbmm
|
||
mindspore.ops.batch_dot
|
||
mindspore.ops.bmm
|
||
mindspore.ops.cholesky
|
||
mindspore.ops.cholesky_solve
|
||
mindspore.ops.cond
|
||
mindspore.ops.dot
|
||
mindspore.ops.eigvals
|
||
mindspore.ops.geqrf
|
||
mindspore.ops.ger
|
||
mindspore.ops.inner
|
||
mindspore.ops.inverse
|
||
mindspore.ops.kron
|
||
mindspore.ops.logdet
|
||
mindspore.ops.lu_solve
|
||
mindspore.ops.lu_unpack
|
||
mindspore.ops.matmul
|
||
mindspore.ops.matrix_band_part
|
||
mindspore.ops.matrix_solve
|
||
mindspore.ops.matrix_diag
|
||
mindspore.ops.matrix_diag_part
|
||
mindspore.ops.matrix_set_diag
|
||
mindspore.ops.mm
|
||
mindspore.ops.mv
|
||
mindspore.ops.outer
|
||
mindspore.ops.orgqr
|
||
mindspore.ops.ormqr
|
||
mindspore.ops.pinv
|
||
mindspore.ops.svd
|
||
mindspore.ops.slogdet
|
||
mindspore.ops.trace
|
||
mindspore.ops.tensor_dot
|
||
mindspore.ops.vander
|
||
mindspore.ops.vecdot
|
||
|
||
谱函数
|
||
^^^^^^^^^^^^^
|
||
|
||
.. mscnplatformautosummary::
|
||
:toctree: ops
|
||
:nosignatures:
|
||
:template: classtemplate.rst
|
||
|
||
mindspore.ops.bartlett_window
|
||
mindspore.ops.blackman_window
|
||
mindspore.ops.hamming_window
|
||
mindspore.ops.hann_window
|
||
mindspore.ops.kaiser_window
|
||
|
||
神经网络层函数
|
||
----------------
|
||
|
||
神经网络
|
||
^^^^^^^^
|
||
|
||
.. mscnplatformautosummary::
|
||
:toctree: ops
|
||
:nosignatures:
|
||
:template: classtemplate.rst
|
||
|
||
mindspore.ops.adaptive_avg_pool1d
|
||
mindspore.ops.adaptive_avg_pool2d
|
||
mindspore.ops.adaptive_avg_pool3d
|
||
mindspore.ops.adaptive_max_pool1d
|
||
mindspore.ops.adaptive_max_pool2d
|
||
mindspore.ops.avg_pool1d
|
||
mindspore.ops.avg_pool2d
|
||
mindspore.ops.avg_pool3d
|
||
mindspore.ops.batch_norm
|
||
mindspore.ops.bias_add
|
||
mindspore.ops.bidense
|
||
mindspore.ops.conv1d
|
||
mindspore.ops.conv2d
|
||
mindspore.ops.conv3d
|
||
mindspore.ops.ctc_greedy_decoder
|
||
mindspore.ops.deformable_conv2d
|
||
mindspore.ops.dense
|
||
mindspore.ops.dropout
|
||
mindspore.ops.dropout1d
|
||
mindspore.ops.dropout2d
|
||
mindspore.ops.dropout3d
|
||
mindspore.ops.embedding
|
||
mindspore.ops.flash_attention_score
|
||
mindspore.ops.flatten
|
||
mindspore.ops.fold
|
||
mindspore.ops.fractional_max_pool3d
|
||
mindspore.ops.fused_infer_attention_score
|
||
mindspore.ops.group_norm
|
||
mindspore.ops.incre_flash_attention
|
||
mindspore.ops.layer_norm
|
||
mindspore.ops.lp_pool1d
|
||
mindspore.ops.lp_pool2d
|
||
mindspore.ops.lrn
|
||
mindspore.ops.max_pool2d
|
||
mindspore.ops.max_pool3d
|
||
mindspore.ops.max_unpool1d
|
||
mindspore.ops.max_unpool2d
|
||
mindspore.ops.max_unpool3d
|
||
mindspore.ops.moe_token_permute
|
||
mindspore.ops.moe_token_unpermute
|
||
mindspore.ops.nsa_select_attention
|
||
mindspore.ops.prompt_flash_attention
|
||
mindspore.ops.ring_attention_update
|
||
mindspore.ops.rms_norm
|
||
mindspore.ops.speed_fusion_attention
|
||
mindspore.ops.unfold
|
||
|
||
损失函数
|
||
^^^^^^^^^^
|
||
|
||
.. mscnplatformautosummary::
|
||
:toctree: ops
|
||
:nosignatures:
|
||
:template: classtemplate.rst
|
||
|
||
mindspore.ops.binary_cross_entropy
|
||
mindspore.ops.binary_cross_entropy_with_logits
|
||
mindspore.ops.cosine_embedding_loss
|
||
mindspore.ops.cross_entropy
|
||
mindspore.ops.ctc_loss
|
||
mindspore.ops.gaussian_nll_loss
|
||
mindspore.ops.hinge_embedding_loss
|
||
mindspore.ops.huber_loss
|
||
mindspore.ops.kl_div
|
||
mindspore.ops.l1_loss
|
||
mindspore.ops.margin_ranking_loss
|
||
mindspore.ops.mse_loss
|
||
mindspore.ops.multi_margin_loss
|
||
mindspore.ops.multilabel_margin_loss
|
||
mindspore.ops.multilabel_soft_margin_loss
|
||
mindspore.ops.nll_loss
|
||
mindspore.ops.smooth_l1_loss
|
||
mindspore.ops.soft_margin_loss
|
||
mindspore.ops.triplet_margin_loss
|
||
|
||
激活函数
|
||
^^^^^^^^^^
|
||
|
||
.. mscnplatformautosummary::
|
||
:toctree: ops
|
||
:nosignatures:
|
||
:template: classtemplate.rst
|
||
|
||
mindspore.ops.celu
|
||
mindspore.ops.elu
|
||
mindspore.ops.fast_gelu
|
||
mindspore.ops.gelu
|
||
mindspore.ops.glu
|
||
mindspore.ops.gumbel_softmax
|
||
mindspore.ops.hardshrink
|
||
mindspore.ops.hardsigmoid
|
||
mindspore.ops.hardswish
|
||
mindspore.ops.hardtanh
|
||
mindspore.ops.leaky_relu
|
||
mindspore.ops.log_softmax
|
||
mindspore.ops.logsigmoid
|
||
mindspore.ops.mish
|
||
mindspore.ops.prelu
|
||
mindspore.ops.relu
|
||
mindspore.ops.relu6
|
||
mindspore.ops.rrelu
|
||
mindspore.ops.selu
|
||
mindspore.ops.sigmoid
|
||
mindspore.ops.silu
|
||
mindspore.ops.softmax
|
||
mindspore.ops.softmin
|
||
mindspore.ops.softshrink
|
||
mindspore.ops.softsign
|
||
mindspore.ops.swiglu
|
||
mindspore.ops.tanh
|
||
mindspore.ops.threshold
|
||
|
||
距离函数
|
||
^^^^^^^^^^
|
||
|
||
.. mscnplatformautosummary::
|
||
:toctree: ops
|
||
:nosignatures:
|
||
:template: classtemplate.rst
|
||
|
||
mindspore.ops.cdist
|
||
mindspore.ops.dist
|
||
mindspore.ops.pdist
|
||
|
||
采样函数
|
||
^^^^^^^^^^
|
||
|
||
.. mscnplatformautosummary::
|
||
:toctree: ops
|
||
:nosignatures:
|
||
:template: classtemplate.rst
|
||
|
||
mindspore.ops.choice_with_mask
|
||
mindspore.ops.random_categorical
|
||
mindspore.ops.log_uniform_candidate_sampler
|
||
mindspore.ops.uniform_candidate_sampler
|
||
|
||
图像函数
|
||
^^^^^^^^^^
|
||
|
||
.. mscnplatformautosummary::
|
||
:toctree: ops
|
||
:nosignatures:
|
||
:template: classtemplate.rst
|
||
|
||
mindspore.ops.affine_grid
|
||
mindspore.ops.bounding_box_decode
|
||
mindspore.ops.bounding_box_encode
|
||
mindspore.ops.col2im
|
||
mindspore.ops.check_valid
|
||
mindspore.ops.crop_and_resize
|
||
mindspore.ops.grid_sample
|
||
mindspore.ops.interpolate
|
||
mindspore.ops.iou
|
||
mindspore.ops.pad
|
||
mindspore.ops.padding
|
||
mindspore.ops.pixel_shuffle
|
||
mindspore.ops.pixel_unshuffle
|
||
mindspore.ops.rotary_position_embedding
|
||
mindspore.ops.rotated_iou
|
||
mindspore.ops.upsample
|
||
|
||
Parameter操作函数
|
||
--------------------
|
||
|
||
.. mscnplatformautosummary::
|
||
:toctree: ops
|
||
:nosignatures:
|
||
:template: classtemplate.rst
|
||
|
||
mindspore.ops.assign
|
||
mindspore.ops.assign_add
|
||
mindspore.ops.assign_sub
|
||
mindspore.ops.scatter_add
|
||
mindspore.ops.scatter_div
|
||
mindspore.ops.scatter_max
|
||
mindspore.ops.scatter_min
|
||
mindspore.ops.scatter_mul
|
||
mindspore.ops.scatter_nd_add
|
||
mindspore.ops.scatter_nd_div
|
||
mindspore.ops.scatter_nd_max
|
||
mindspore.ops.scatter_nd_min
|
||
mindspore.ops.scatter_nd_mul
|
||
mindspore.ops.scatter_nd_sub
|
||
mindspore.ops.scatter_update
|
||
|
||
微分函数
|
||
----------------
|
||
|
||
.. mscnplatformautosummary::
|
||
:toctree: ops
|
||
:nosignatures:
|
||
:template: classtemplate.rst
|
||
|
||
mindspore.ops.derivative
|
||
mindspore.ops.jet
|
||
mindspore.ops.stop_gradient
|
||
|
||
调试函数
|
||
----------------
|
||
|
||
.. mscnplatformautosummary::
|
||
:toctree: ops
|
||
:nosignatures:
|
||
:template: classtemplate.rst
|
||
|
||
mindspore.ops.print_
|
||
mindspore.ops.tensordump
|
||
|
||
稀疏函数
|
||
----------------
|
||
|
||
.. warning::
|
||
这些是实验性API，后续可能修改或删除。
|
||
|
||
.. mscnplatformautosummary::
|
||
:toctree: ops
|
||
:nosignatures:
|
||
:template: classtemplate.rst
|
||
|
||
mindspore.ops.dense_to_sparse_coo
|
||
mindspore.ops.dense_to_sparse_csr
|
||
mindspore.ops.csr_to_coo
|
||
|
||
COO函数
|
||
^^^^^^^^^^^^^^^^
|
||
|
||
.. warning::
|
||
这些是实验性API，后续可能修改或删除。
|
||
|
||
.. mscnplatformautosummary::
|
||
:toctree: ops
|
||
:nosignatures:
|
||
:template: classtemplate.rst
|
||
|
||
mindspore.ops.coo_abs
|
||
mindspore.ops.coo_acos
|
||
mindspore.ops.coo_acosh
|
||
mindspore.ops.coo_add
|
||
mindspore.ops.coo_asin
|
||
mindspore.ops.coo_asinh
|
||
mindspore.ops.coo_atan
|
||
mindspore.ops.coo_atanh
|
||
mindspore.ops.coo_ceil
|
||
mindspore.ops.coo_concat
|
||
mindspore.ops.coo_cos
|
||
mindspore.ops.coo_cosh
|
||
mindspore.ops.coo_exp
|
||
mindspore.ops.coo_expm1
|
||
mindspore.ops.coo_floor
|
||
mindspore.ops.coo_inv
|
||
mindspore.ops.coo_isfinite
|
||
mindspore.ops.coo_isinf
|
||
mindspore.ops.coo_isnan
|
||
mindspore.ops.coo_log
|
||
mindspore.ops.coo_log1p
|
||
mindspore.ops.coo_neg
|
||
mindspore.ops.coo_relu
|
||
mindspore.ops.coo_relu6
|
||
mindspore.ops.coo_round
|
||
mindspore.ops.coo_sigmoid
|
||
mindspore.ops.coo_sin
|
||
mindspore.ops.coo_sinh
|
||
mindspore.ops.coo_softsign
|
||
mindspore.ops.coo_sqrt
|
||
mindspore.ops.coo_square
|
||
mindspore.ops.coo_tan
|
||
mindspore.ops.coo_tanh
|
||
|
||
CSR函数
|
||
^^^^^^^^^^^^^^^^
|
||
|
||
.. warning::
|
||
这些是实验性API，后续可能修改或删除。
|
||
|
||
.. mscnplatformautosummary::
|
||
:toctree: ops
|
||
:nosignatures:
|
||
:template: classtemplate.rst
|
||
|
||
mindspore.ops.csr_abs
|
||
mindspore.ops.csr_acos
|
||
mindspore.ops.csr_acosh
|
||
mindspore.ops.csr_add
|
||
mindspore.ops.csr_asin
|
||
mindspore.ops.csr_asinh
|
||
mindspore.ops.csr_atan
|
||
mindspore.ops.csr_atanh
|
||
mindspore.ops.csr_ceil
|
||
mindspore.ops.csr_cos
|
||
mindspore.ops.csr_cosh
|
||
mindspore.ops.csr_exp
|
||
mindspore.ops.csr_expm1
|
||
mindspore.ops.csr_floor
|
||
mindspore.ops.csr_inv
|
||
mindspore.ops.csr_isfinite
|
||
mindspore.ops.csr_isinf
|
||
mindspore.ops.csr_isnan
|
||
mindspore.ops.csr_log
|
||
mindspore.ops.csr_log1p
|
||
mindspore.ops.csr_mm
|
||
mindspore.ops.csr_neg
|
||
mindspore.ops.csr_relu
|
||
mindspore.ops.csr_relu6
|
||
mindspore.ops.csr_round
|
||
mindspore.ops.csr_sigmoid
|
||
mindspore.ops.csr_sin
|
||
mindspore.ops.csr_sinh
|
||
mindspore.ops.csr_softmax
|
||
mindspore.ops.csr_softsign
|
||
mindspore.ops.csr_sqrt
|
||
mindspore.ops.csr_square
|
||
mindspore.ops.csr_tan
|
||
mindspore.ops.csr_tanh
|
||
|
||
MC2 函数
|
||
----------------
|
||
|
||
.. warning::
|
||
这些是实验性API，后续可能修改或删除。
|
||
|
||
.. mscnplatformautosummary::
|
||
:toctree: ops
|
||
:nosignatures:
|
||
:template: classtemplate.rst
|
||
|
||
mindspore.ops.all_gather_matmul
|
||
mindspore.ops.matmul_reduce_scatter
|