forked from mindspore-Ecosystem/mindspore
485 lines
11 KiB
ReStructuredText
485 lines
11 KiB
ReStructuredText
mindspore.ops.function
|
|
=============================
|
|
|
|
神经网络层函数
|
|
----------------
|
|
|
|
神经网络
|
|
^^^^^^^^
|
|
|
|
.. 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_pool3d
|
|
mindspore.ops.avg_pool1d
|
|
mindspore.ops.avg_pool2d
|
|
mindspore.ops.avg_pool3d
|
|
mindspore.ops.batch_norm
|
|
mindspore.ops.bias_add
|
|
mindspore.ops.ctc_greedy_decoder
|
|
mindspore.ops.conv2d
|
|
mindspore.ops.conv3d
|
|
mindspore.ops.deformable_conv2d
|
|
mindspore.ops.dropout2d
|
|
mindspore.ops.dropout3d
|
|
mindspore.ops.flatten
|
|
mindspore.ops.interpolate
|
|
mindspore.ops.lrn
|
|
mindspore.ops.max_pool3d
|
|
mindspore.ops.kl_div
|
|
mindspore.ops.pad
|
|
mindspore.ops.padding
|
|
mindspore.ops.pdist
|
|
mindspore.ops.prelu
|
|
mindspore.ops.relu
|
|
mindspore.ops.relu6
|
|
|
|
损失函数
|
|
^^^^^^^^^^
|
|
|
|
.. mscnplatformautosummary::
|
|
:toctree: ops
|
|
:nosignatures:
|
|
:template: classtemplate.rst
|
|
|
|
mindspore.ops.binary_cross_entropy
|
|
mindspore.ops.binary_cross_entropy_with_logits
|
|
mindspore.ops.cross_entropy
|
|
mindspore.ops.nll_loss
|
|
mindspore.ops.smooth_l1_loss
|
|
|
|
激活函数
|
|
^^^^^^^^^^
|
|
|
|
.. mscnplatformautosummary::
|
|
:toctree: ops
|
|
:nosignatures:
|
|
:template: classtemplate.rst
|
|
|
|
mindspore.ops.celu
|
|
mindspore.ops.dropout
|
|
mindspore.ops.fast_gelu
|
|
mindspore.ops.gumbel_softmax
|
|
mindspore.ops.hardshrink
|
|
mindspore.ops.hardswish
|
|
mindspore.ops.log_softmax
|
|
mindspore.ops.mish
|
|
mindspore.ops.selu
|
|
mindspore.ops.soft_shrink
|
|
mindspore.ops.softmax
|
|
mindspore.ops.softsign
|
|
mindspore.ops.tanh
|
|
|
|
采样函数
|
|
^^^^^^^^^^
|
|
|
|
.. mscnplatformautosummary::
|
|
:toctree: ops
|
|
:nosignatures:
|
|
:template: classtemplate.rst
|
|
|
|
mindspore.ops.grid_sample
|
|
mindspore.ops.uniform_candidate_sampler
|
|
|
|
距离函数
|
|
^^^^^^^^^^
|
|
|
|
.. mscnplatformautosummary::
|
|
:toctree: ops
|
|
:nosignatures:
|
|
:template: classtemplate.rst
|
|
|
|
mindspore.ops.cdist
|
|
|
|
数学运算函数
|
|
^^^^^^^^^^^^^^^^^
|
|
|
|
.. mscnplatformautosummary::
|
|
:toctree: ops
|
|
:nosignatures:
|
|
:template: classtemplate.rst
|
|
|
|
mindspore.ops.bmm
|
|
mindspore.ops.cholesky
|
|
mindspore.ops.cholesky_inverse
|
|
|
|
逐元素运算
|
|
^^^^^^^^^^^^^
|
|
|
|
.. mscnplatformautosummary::
|
|
:toctree: ops
|
|
:nosignatures:
|
|
:template: classtemplate.rst
|
|
|
|
mindspore.ops.abs
|
|
mindspore.ops.accumulate_n
|
|
mindspore.ops.acos
|
|
mindspore.ops.acosh
|
|
mindspore.ops.add
|
|
mindspore.ops.addn
|
|
mindspore.ops.asin
|
|
mindspore.ops.asinh
|
|
mindspore.ops.atan
|
|
mindspore.ops.atan2
|
|
mindspore.ops.atanh
|
|
mindspore.ops.bernoulli
|
|
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_or
|
|
mindspore.ops.bitwise_xor
|
|
mindspore.ops.ceil
|
|
mindspore.ops.cos
|
|
mindspore.ops.cosh
|
|
mindspore.ops.div
|
|
mindspore.ops.erf
|
|
mindspore.ops.erfc
|
|
mindspore.ops.exp
|
|
mindspore.ops.expm1
|
|
mindspore.ops.floor
|
|
mindspore.ops.floor_div
|
|
mindspore.ops.floor_mod
|
|
mindspore.ops.inv
|
|
mindspore.ops.invert
|
|
mindspore.ops.lerp
|
|
mindspore.ops.log
|
|
mindspore.ops.log1p
|
|
mindspore.ops.logical_and
|
|
mindspore.ops.logical_not
|
|
mindspore.ops.logical_or
|
|
mindspore.ops.logit
|
|
mindspore.ops.log_matrix_determinant
|
|
mindspore.ops.matrix_determinant
|
|
mindspore.ops.mul
|
|
mindspore.ops.neg
|
|
mindspore.ops.pow
|
|
mindspore.ops.round
|
|
mindspore.ops.sin
|
|
mindspore.ops.sinh
|
|
mindspore.ops.sqrt
|
|
mindspore.ops.square
|
|
mindspore.ops.sub
|
|
mindspore.ops.svd
|
|
mindspore.ops.tan
|
|
mindspore.ops.trunc
|
|
mindspore.ops.truncate_div
|
|
mindspore.ops.truncate_mod
|
|
mindspore.ops.xdivy
|
|
mindspore.ops.xlogy
|
|
|
|
Reduction函数
|
|
^^^^^^^^^^^^^
|
|
|
|
.. mscnplatformautosummary::
|
|
:toctree: ops
|
|
:nosignatures:
|
|
:template: classtemplate.rst
|
|
|
|
mindspore.ops.amax
|
|
mindspore.ops.amin
|
|
mindspore.ops.argmax
|
|
mindspore.ops.argmin
|
|
mindspore.ops.cummax
|
|
mindspore.ops.cummin
|
|
mindspore.ops.logsumexp
|
|
mindspore.ops.max
|
|
mindspore.ops.mean
|
|
mindspore.ops.median
|
|
mindspore.ops.min
|
|
mindspore.ops.norm
|
|
mindspore.ops.prod
|
|
mindspore.ops.std
|
|
|
|
比较函数
|
|
^^^^^^^^^^^^^
|
|
|
|
.. mscnplatformautosummary::
|
|
:toctree: ops
|
|
:nosignatures:
|
|
:template: classtemplate.rst
|
|
|
|
mindspore.ops.approximate_equal
|
|
mindspore.ops.equal
|
|
mindspore.ops.ge
|
|
mindspore.ops.gt
|
|
mindspore.ops.intopk
|
|
mindspore.ops.isclose
|
|
mindspore.ops.isfinite
|
|
mindspore.ops.isnan
|
|
mindspore.ops.le
|
|
mindspore.ops.less
|
|
mindspore.ops.maximum
|
|
mindspore.ops.minimum
|
|
mindspore.ops.ne
|
|
mindspore.ops.same_type_shape
|
|
|
|
线性代数函数
|
|
^^^^^^^^^^^^^
|
|
|
|
.. mscnplatformautosummary::
|
|
:toctree: ops
|
|
:nosignatures:
|
|
:template: classtemplate.rst
|
|
|
|
mindspore.ops.batch_dot
|
|
mindspore.ops.dot
|
|
mindspore.ops.matmul
|
|
mindspore.ops.matrix_solve
|
|
mindspore.ops.ger
|
|
mindspore.ops.renorm
|
|
mindspore.ops.tensor_dot
|
|
|
|
Tensor操作函数
|
|
----------------
|
|
|
|
Tensor创建
|
|
^^^^^^^^^^^^^
|
|
|
|
.. mscnplatformautosummary::
|
|
:toctree: ops
|
|
:nosignatures:
|
|
:template: classtemplate.rst
|
|
|
|
mindspore.ops.eye
|
|
mindspore.ops.fill
|
|
mindspore.ops.fills
|
|
mindspore.ops.linspace
|
|
mindspore.ops.narrow
|
|
mindspore.ops.one_hot
|
|
mindspore.ops.ones
|
|
mindspore.ops.ones_like
|
|
mindspore.ops.zeros_like
|
|
|
|
随机生成函数
|
|
^^^^^^^^^^^^^^^^
|
|
|
|
.. mscnplatformautosummary::
|
|
:toctree: ops
|
|
:nosignatures:
|
|
:template: classtemplate.rst
|
|
|
|
mindspore.ops.choice_with_mask
|
|
mindspore.ops.gamma
|
|
mindspore.ops.laplace
|
|
mindspore.ops.multinomial
|
|
mindspore.ops.poisson
|
|
mindspore.ops.random_poisson
|
|
mindspore.ops.random_categorical
|
|
mindspore.ops.random_gamma
|
|
mindspore.ops.standard_laplace
|
|
mindspore.ops.standard_normal
|
|
mindspore.ops.uniform
|
|
|
|
Array操作
|
|
^^^^^^^^^^^^^^^^
|
|
|
|
.. mscnplatformautosummary::
|
|
:toctree: ops
|
|
:nosignatures:
|
|
:template: classtemplate.rst
|
|
|
|
mindspore.ops.adaptive_max_pool2d
|
|
mindspore.ops.affine_grid
|
|
mindspore.ops.batch_to_space_nd
|
|
mindspore.ops.broadcast_to
|
|
mindspore.ops.col2im
|
|
mindspore.ops.concat
|
|
mindspore.ops.count_nonzero
|
|
mindspore.ops.diag
|
|
mindspore.ops.dyn_shape
|
|
mindspore.ops.expand
|
|
mindspore.ops.expand_dims
|
|
mindspore.ops.gather
|
|
mindspore.ops.gather_d
|
|
mindspore.ops.gather_elements
|
|
mindspore.ops.gather_nd
|
|
mindspore.ops.index_add
|
|
mindspore.ops.index_fill
|
|
mindspore.ops.inplace_add
|
|
mindspore.ops.inplace_sub
|
|
mindspore.ops.inplace_update
|
|
mindspore.ops.masked_fill
|
|
mindspore.ops.masked_select
|
|
mindspore.ops.matrix_band_part
|
|
mindspore.ops.matrix_diag
|
|
mindspore.ops.matrix_diag_part
|
|
mindspore.ops.matrix_set_diag
|
|
mindspore.ops.meshgrid
|
|
mindspore.ops.normal
|
|
mindspore.ops.nonzero
|
|
mindspore.ops.population_count
|
|
mindspore.ops.range
|
|
mindspore.ops.rank
|
|
mindspore.ops.repeat_elements
|
|
mindspore.ops.reshape
|
|
mindspore.ops.reverse_sequence
|
|
mindspore.ops.scatter_nd
|
|
mindspore.ops.select
|
|
mindspore.ops.sequence_mask
|
|
mindspore.ops.shape
|
|
mindspore.ops.size
|
|
mindspore.ops.slice
|
|
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.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.tile
|
|
mindspore.ops.top_k
|
|
mindspore.ops.transpose
|
|
mindspore.ops.unique
|
|
mindspore.ops.unique_consecutive
|
|
mindspore.ops.unique_with_pad
|
|
mindspore.ops.unsorted_segment_max
|
|
mindspore.ops.unsorted_segment_min
|
|
mindspore.ops.unsorted_segment_prod
|
|
mindspore.ops.unsorted_segment_sum
|
|
mindspore.ops.unstack
|
|
|
|
类型转换
|
|
^^^^^^^^^^^^^^^^
|
|
|
|
.. mscnplatformautosummary::
|
|
:toctree: ops
|
|
:nosignatures:
|
|
:template: classtemplate.rst
|
|
|
|
mindspore.ops.scalar_cast
|
|
mindspore.ops.scalar_to_array
|
|
mindspore.ops.scalar_to_tensor
|
|
mindspore.ops.tuple_to_array
|
|
|
|
稀疏函数
|
|
^^^^^^^^^^^^^^^^
|
|
|
|
.. mscnplatformautosummary::
|
|
:toctree: ops
|
|
:nosignatures:
|
|
:template: classtemplate.rst
|
|
|
|
mindspore.ops.dense_to_sparse_coo
|
|
mindspore.ops.dense_to_sparse_csr
|
|
mindspore.ops.csr_add
|
|
mindspore.ops.csr_softmax
|
|
mindspore.ops.csr_to_coo
|
|
mindspore.ops.sparse_add
|
|
|
|
梯度剪裁
|
|
^^^^^^^^
|
|
|
|
.. mscnplatformautosummary::
|
|
:toctree: ops
|
|
:nosignatures:
|
|
:template: classtemplate.rst
|
|
|
|
mindspore.ops.clip_by_global_norm
|
|
mindspore.ops.clip_by_value
|
|
|
|
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.grad
|
|
mindspore.ops.value_and_grad
|
|
mindspore.ops.jet
|
|
mindspore.ops.jvp
|
|
mindspore.ops.vjp
|
|
mindspore.ops.jacfwd
|
|
mindspore.ops.vmap
|
|
|
|
调试函数
|
|
----------------
|
|
|
|
.. mscnplatformautosummary::
|
|
:toctree: ops
|
|
:nosignatures:
|
|
:template: classtemplate.rst
|
|
|
|
mindspore.ops.print_
|
|
|
|
图像函数
|
|
----------------
|
|
|
|
.. msplatformautosummary::
|
|
:toctree: ops
|
|
:nosignatures:
|
|
:template: classtemplate.rst
|
|
|
|
mindspore.ops.bounding_box_decode
|
|
mindspore.ops.bounding_box_encode
|
|
mindspore.ops.check_valid
|
|
mindspore.ops.iou
|
|
|
|
光谱函数
|
|
----------------
|
|
|
|
.. msplatformautosummary::
|
|
:toctree: ops
|
|
:nosignatures:
|
|
:template: classtemplate.rst
|
|
|
|
mindspore.ops.blackman_window
|
|
|
|
其他函数
|
|
----------------
|
|
|
|
.. mscnplatformautosummary::
|
|
:toctree: ops
|
|
:nosignatures:
|
|
:template: classtemplate.rst
|
|
|
|
mindspore.ops.core
|