FT_DSP.gitlink.net/master/html/functionlib/dsplib/switch.html

472 lines
47 KiB
HTML
Raw Blame History

This file contains ambiguous Unicode characters

This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.

<!DOCTYPE html>
<html class="writer-html5" lang="zh-CN">
<head>
<meta charset="utf-8" /><meta name="viewport" content="width=device-width, initial-scale=1" />
<meta name="viewport" content="width=device-width, initial-scale=1.0" />
<title>Switch &mdash; MindSpore Signal+ 使用手册 alpha 文档</title>
<link rel="stylesheet" type="text/css" href="../../_static/pygments.css?v=03e43079" />
<link rel="stylesheet" type="text/css" href="../../_static/css/theme.css?v=e59714d7" />
<link rel="stylesheet" type="text/css" href="../../_static/table.css?v=72b86d3e" />
<link rel="stylesheet" type="text/css" href="../../_static/pageviews.css?v=ff0ce7dc" />
<script src="../../_static/jquery.js?v=5d32c60e"></script>
<script src="../../_static/_sphinx_javascript_frameworks_compat.js?v=2cd50e6c"></script>
<script data-url_root="../../" id="documentation_options" src="../../_static/documentation_options.js?v=595569a8"></script>
<script src="../../_static/doctools.js?v=888ff710"></script>
<script src="../../_static/sphinx_highlight.js?v=4825356b"></script>
<script src="../../_static/translations.js?v=beaddf03"></script>
<script async="async" src="https://cdn.jsdelivr.net/npm/mathjax@3/es5/tex-mml-chtml.js"></script>
<script src="../../_static/table.js?v=99ed15ba"></script>
<script src="../../_static/js/theme.js"></script>
<link rel="index" title="索引" href="../../genindex.html" />
<link rel="search" title="搜索" href="../../search.html" />
<link rel="next" title="Switchlayer" href="switchlayer.html" />
<link rel="prev" title="SubGrad" href="sub_grad.html" />
</head>
<body class="wy-body-for-nav">
<div class="wy-grid-for-nav">
<nav data-toggle="wy-nav-shift" class="wy-nav-side">
<div class="wy-side-scroll">
<div class="wy-side-nav-search" >
<a href="../../index.html" class="icon icon-home">
MindSpore Signal+ 使用手册
</a>
<div role="search">
<form id="rtd-search-form" class="wy-form" action="../../search.html" method="get">
<input type="text" name="q" placeholder="搜索文档" aria-label="搜索文档" />
<input type="hidden" name="check_keywords" value="yes" />
<input type="hidden" name="area" value="default" />
</form>
</div>
</div><div class="wy-menu wy-menu-vertical" data-spy="affix" role="navigation" aria-label="导航菜单">
<ul class="current">
<li class="toctree-l1"><a class="reference internal" href="../../quickstart/index.html">快速入门</a></li>
<li class="toctree-l1"><a class="reference internal" href="../../appdevelop/index.html">应用开发示例</a></li>
<li class="toctree-l1 current"><a class="reference internal" href="../index.html">算子库支持</a><ul class="current">
<li class="toctree-l2"><a class="reference internal" href="../supported_op.html">算子库支持情况</a></li>
<li class="toctree-l2"><a class="reference internal" href="../custom_op/index.html">自定义算子列表</a></li>
<li class="toctree-l2 current"><a class="reference internal" href="dsplib_index.html">DSP Library C API Reference</a><ul class="current">
<li class="toctree-l3"><a class="reference internal" href="abs.html">Abs</a></li>
<li class="toctree-l3"><a class="reference internal" href="abs_grad.html">AbsGrad</a></li>
<li class="toctree-l3"><a class="reference internal" href="activation.html">Activation</a></li>
<li class="toctree-l3"><a class="reference internal" href="activation_grad.html">ActivationGrad</a></li>
<li class="toctree-l3"><a class="reference internal" href="adam.html">Adam</a></li>
<li class="toctree-l3"><a class="reference internal" href="adamweightdecay.html">AdamWeightDecay</a></li>
<li class="toctree-l3"><a class="reference internal" href="adder.html">AdderFusion</a></li>
<li class="toctree-l3"><a class="reference internal" href="add.html">AddFusion</a></li>
<li class="toctree-l3"><a class="reference internal" href="add_grad.html">AddGrad</a></li>
<li class="toctree-l3"><a class="reference internal" href="addn.html">AddN</a></li>
<li class="toctree-l3"><a class="reference internal" href="affine.html">Affine</a></li>
<li class="toctree-l3"><a class="reference internal" href="all.html">All</a></li>
<li class="toctree-l3"><a class="reference internal" href="allgather.html">AllGather</a></li>
<li class="toctree-l3"><a class="reference internal" href="apply_momentum.html">ApplyMomentum</a></li>
<li class="toctree-l3"><a class="reference internal" href="argmax.html">Argmax</a></li>
<li class="toctree-l3"><a class="reference internal" href="argmin.html">Argmin</a></li>
<li class="toctree-l3"><a class="reference internal" href="assert.html">Assert</a></li>
<li class="toctree-l3"><a class="reference internal" href="assign.html">Assign</a></li>
<li class="toctree-l3"><a class="reference internal" href="assignadd.html">AssignAdd</a></li>
<li class="toctree-l3"><a class="reference internal" href="attention.html">Attention</a></li>
<li class="toctree-l3"><a class="reference internal" href="audio_spectrogram.html">AudioSpectrogram</a></li>
<li class="toctree-l3"><a class="reference internal" href="avgpooling.html">Avgpooling</a></li>
<li class="toctree-l3"><a class="reference internal" href="avg_pool_grad.html">AvgPoolGrad</a></li>
<li class="toctree-l3"><a class="reference internal" href="batchnorm.html">BatchNorm</a></li>
<li class="toctree-l3"><a class="reference internal" href="batchnormgrad.html">Batchnormgrad</a></li>
<li class="toctree-l3"><a class="reference internal" href="batch_to_space.html">BatchToSpace</a></li>
<li class="toctree-l3"><a class="reference internal" href="batchtospacend.html">BatchToSpaceND</a></li>
<li class="toctree-l3"><a class="reference internal" href="biasadd.html">Biasadd</a></li>
<li class="toctree-l3"><a class="reference internal" href="biasaddgrad.html">Biasaddgrad</a></li>
<li class="toctree-l3"><a class="reference internal" href="binarycrossentropy.html">Binarycrossentropy</a></li>
<li class="toctree-l3"><a class="reference internal" href="binary_cross_entropy_grad.html">BinaryCrossEntropyGrad</a></li>
<li class="toctree-l3"><a class="reference internal" href="broadcastto.html">BroadcastTo</a></li>
<li class="toctree-l3"><a class="reference internal" href="cast.html">Cast</a></li>
<li class="toctree-l3"><a class="reference internal" href="ceil.html">Ceil</a></li>
<li class="toctree-l3"><a class="reference internal" href="clip.html">Clip</a></li>
<li class="toctree-l3"><a class="reference internal" href="concat.html">Concat</a></li>
<li class="toctree-l3"><a class="reference internal" href="constant_of_shape.html">ConstantOfShape</a></li>
<li class="toctree-l3"><a class="reference internal" href="conv2d.html">Conv2d</a></li>
<li class="toctree-l3"><a class="reference internal" href="conv2d_transpose.html">Conv2dTranspose</a></li>
<li class="toctree-l3"><a class="reference internal" href="conv2dbackpropfilterfusion.html">Conv2DBackpropFilterFusion</a></li>
<li class="toctree-l3"><a class="reference internal" href="conv2dbackpropinputfusion.html">Conv2DBackpropInputFusion</a></li>
<li class="toctree-l3"><a class="reference internal" href="cos.html">Cos</a></li>
<li class="toctree-l3"><a class="reference internal" href="crop.html">Crop</a></li>
<li class="toctree-l3"><a class="reference internal" href="crop_and_resize.html">CropAndResize</a></li>
<li class="toctree-l3"><a class="reference internal" href="cumsum.html">Cumsum</a></li>
<li class="toctree-l3"><a class="reference internal" href="customextractfeatures.html">CustomExtractFeatures</a></li>
<li class="toctree-l3"><a class="reference internal" href="customnormalize.html">Customnormalize</a></li>
<li class="toctree-l3"><a class="reference internal" href="custompredict.html">CustomPredict</a></li>
<li class="toctree-l3"><a class="reference internal" href="deconvgradfilter.html">DeconvGradFilter</a></li>
<li class="toctree-l3"><a class="reference internal" href="depthtospace.html">DepthToSpace</a></li>
<li class="toctree-l3"><a class="reference internal" href="detection_post_process.html">DetectionPostProcess</a></li>
<li class="toctree-l3"><a class="reference internal" href="div.html">DivFusion</a></li>
<li class="toctree-l3"><a class="reference internal" href="divgrad.html">Divgrad</a></li>
<li class="toctree-l3"><a class="reference internal" href="dropout.html">Dropout</a></li>
<li class="toctree-l3"><a class="reference internal" href="dropoutgrad.html">Dropoutgrad</a></li>
<li class="toctree-l3"><a class="reference internal" href="dynamic_quant.html">DynamicQuant</a></li>
<li class="toctree-l3"><a class="reference internal" href="eltwise.html">Eltwise</a></li>
<li class="toctree-l3"><a class="reference internal" href="elu.html">Elu</a></li>
<li class="toctree-l3"><a class="reference internal" href="embeddinglookup.html">EmbeddingLookup</a></li>
<li class="toctree-l3"><a class="reference internal" href="equal.html">Equal</a></li>
<li class="toctree-l3"><a class="reference internal" href="erf.html">Erf</a></li>
<li class="toctree-l3"><a class="reference internal" href="expand_dims.html">ExpandDims</a></li>
<li class="toctree-l3"><a class="reference internal" href="expfusion.html">ExpFusion</a></li>
<li class="toctree-l3"><a class="reference internal" href="fake_quant_with_min_max_vars.html">FakeQuantWithMinMaxVars</a></li>
<li class="toctree-l3"><a class="reference internal" href="fake_quant_with_min_max_vars_per_channel.html">FakeQuantWithMinMaxVarsPerChannel</a></li>
<li class="toctree-l3"><a class="reference internal" href="fftimag.html">FFTImag</a></li>
<li class="toctree-l3"><a class="reference internal" href="fftreal.html">FFTReal</a></li>
<li class="toctree-l3"><a class="reference internal" href="fill.html">Fill</a></li>
<li class="toctree-l3"><a class="reference internal" href="fillv2.html">FillV2</a></li>
<li class="toctree-l3"><a class="reference internal" href="flatten.html">Flatten</a></li>
<li class="toctree-l3"><a class="reference internal" href="flattengrad.html">FlattenGrad</a></li>
<li class="toctree-l3"><a class="reference internal" href="floor.html">Floor</a></li>
<li class="toctree-l3"><a class="reference internal" href="floordiv.html">FloorDiv</a></li>
<li class="toctree-l3"><a class="reference internal" href="floor_mod.html">FloorMod</a></li>
<li class="toctree-l3"><a class="reference internal" href="formattranspose.html">FormatTranspose</a></li>
<li class="toctree-l3"><a class="reference internal" href="fullconnection.html">FullConnection</a></li>
<li class="toctree-l3"><a class="reference internal" href="fusedbatchnorm.html">FusedBatchNorm</a></li>
<li class="toctree-l3"><a class="reference internal" href="gather.html">Gather</a></li>
<li class="toctree-l3"><a class="reference internal" href="gatherd.html">GatherD</a></li>
<li class="toctree-l3"><a class="reference internal" href="gather_nd.html">GatherNd</a></li>
<li class="toctree-l3"><a class="reference internal" href="glu.html">GLU</a></li>
<li class="toctree-l3"><a class="reference internal" href="greater.html">Greater</a></li>
<li class="toctree-l3"><a class="reference internal" href="greaterequal.html">GreaterEqual</a></li>
<li class="toctree-l3"><a class="reference internal" href="groupnormfusion.html">GroupNormFusion</a></li>
<li class="toctree-l3"><a class="reference internal" href="gru.html">GRU</a></li>
<li class="toctree-l3"><a class="reference internal" href="hashtablelookup.html">HashtableLookup</a></li>
<li class="toctree-l3"><a class="reference internal" href="instancenorm.html">InstanceNorm</a></li>
<li class="toctree-l3"><a class="reference internal" href="invert_permutation.html">InvertPermutation</a></li>
<li class="toctree-l3"><a class="reference internal" href="isfinite.html">Isfinite</a></li>
<li class="toctree-l3"><a class="reference internal" href="l2norm.html">L2norm</a></li>
<li class="toctree-l3"><a class="reference internal" href="layernormfusion.html">LayerNormFusion</a></li>
<li class="toctree-l3"><a class="reference internal" href="layernormgrad.html">Layernormgrad</a></li>
<li class="toctree-l3"><a class="reference internal" href="leaky_relu.html">LeakyReLu</a></li>
<li class="toctree-l3"><a class="reference internal" href="less.html">Less</a></li>
<li class="toctree-l3"><a class="reference internal" href="lessequal.html">LessEqual</a></li>
<li class="toctree-l3"><a class="reference internal" href="linspace.html">LinSpace</a></li>
<li class="toctree-l3"><a class="reference internal" href="log.html">Log</a></li>
<li class="toctree-l3"><a class="reference internal" href="log1p.html">Log1p</a></li>
<li class="toctree-l3"><a class="reference internal" href="loggrad.html">LogGrad</a></li>
<li class="toctree-l3"><a class="reference internal" href="logical_and.html">LogicalAnd</a></li>
<li class="toctree-l3"><a class="reference internal" href="logical_not.html">LogicalNot</a></li>
<li class="toctree-l3"><a class="reference internal" href="logical_or.html">LogicalOr</a></li>
<li class="toctree-l3"><a class="reference internal" href="logsoftmax.html">LogSoftmax</a></li>
<li class="toctree-l3"><a class="reference internal" href="lpnormalization.html">LpNormalization</a></li>
<li class="toctree-l3"><a class="reference internal" href="lrn.html">Lrn</a></li>
<li class="toctree-l3"><a class="reference internal" href="lsh_projection.html">LshProjection</a></li>
<li class="toctree-l3"><a class="reference internal" href="lstm.html">LSTM</a></li>
<li class="toctree-l3"><a class="reference internal" href="lstmgrad.html">LstmGrad</a></li>
<li class="toctree-l3"><a class="reference internal" href="lstmgraddata.html">LstmGradData</a></li>
<li class="toctree-l3"><a class="reference internal" href="lstmgradweight.html">LstmGradWeight</a></li>
<li class="toctree-l3"><a class="reference internal" href="matmul_fusion.html">MatMulFusion</a></li>
<li class="toctree-l3"><a class="reference internal" href="maximum.html">Maximum</a></li>
<li class="toctree-l3"><a class="reference internal" href="maximumgrad.html">Maximumgrad</a></li>
<li class="toctree-l3"><a class="reference internal" href="maxpoolfusion.html">MaxPoolFusion</a></li>
<li class="toctree-l3"><a class="reference internal" href="maxpoolgrad.html">MaxPoolGrad</a></li>
<li class="toctree-l3"><a class="reference internal" href="mfcc.html">MFCC</a></li>
<li class="toctree-l3"><a class="reference internal" href="minimum.html">Minimum</a></li>
<li class="toctree-l3"><a class="reference internal" href="minimumgrad.html">Minimumgrad</a></li>
<li class="toctree-l3"><a class="reference internal" href="mod.html">Mod</a></li>
<li class="toctree-l3"><a class="reference internal" href="mul.html">MulFusion</a></li>
<li class="toctree-l3"><a class="reference internal" href="mulgrad.html">Mulgrad</a></li>
<li class="toctree-l3"><a class="reference internal" href="neg.html">Neg</a></li>
<li class="toctree-l3"><a class="reference internal" href="neg_grad.html">NegGrad</a></li>
<li class="toctree-l3"><a class="reference internal" href="nllloss.html">NLLLoss</a></li>
<li class="toctree-l3"><a class="reference internal" href="nlllossgrad.html">NLLLossGrad</a></li>
<li class="toctree-l3"><a class="reference internal" href="non_max_suppression.html">NonMaxSuppression</a></li>
<li class="toctree-l3"><a class="reference internal" href="nonzero.html">NonZero</a></li>
<li class="toctree-l3"><a class="reference internal" href="not_equal.html">NotEqual</a></li>
<li class="toctree-l3"><a class="reference internal" href="onehot.html">OneHot</a></li>
<li class="toctree-l3"><a class="reference internal" href="ones_like.html">OnesLike</a></li>
<li class="toctree-l3"><a class="reference internal" href="padfusion.html">PadFusion</a></li>
<li class="toctree-l3"><a class="reference internal" href="power_grad.html">PowerGrad</a></li>
<li class="toctree-l3"><a class="reference internal" href="pow_fusion.html">PowFusion</a></li>
<li class="toctree-l3"><a class="reference internal" href="prelufusion.html">PReLUFusion</a></li>
<li class="toctree-l3"><a class="reference internal" href="priorbox.html">Priorbox</a></li>
<li class="toctree-l3"><a class="reference internal" href="quantdtypecast.html">QuantDTypeCast</a></li>
<li class="toctree-l3"><a class="reference internal" href="ragged_range.html">RaggedRange</a></li>
<li class="toctree-l3"><a class="reference internal" href="random_normal.html">RandomNormal</a></li>
<li class="toctree-l3"><a class="reference internal" href="random_standard_normal.html">RandomStandardNormal</a></li>
<li class="toctree-l3"><a class="reference internal" href="range.html">Range</a></li>
<li class="toctree-l3"><a class="reference internal" href="rank.html">Rank</a></li>
<li class="toctree-l3"><a class="reference internal" href="real_div.html">RealDiv</a></li>
<li class="toctree-l3"><a class="reference internal" href="reciprocal.html">Reciprocal</a></li>
<li class="toctree-l3"><a class="reference internal" href="reduce.html">Reduce</a></li>
<li class="toctree-l3"><a class="reference internal" href="reducescatter.html">ReduceScatter</a></li>
<li class="toctree-l3"><a class="reference internal" href="reshape.html">Reshape</a></li>
<li class="toctree-l3"><a class="reference internal" href="resize.html">Resize</a></li>
<li class="toctree-l3"><a class="reference internal" href="resizegrad.html">Resizegrad</a></li>
<li class="toctree-l3"><a class="reference internal" href="reverse_sequence.html">ReverseSequence</a></li>
<li class="toctree-l3"><a class="reference internal" href="reversev2.html">ReverseV2</a></li>
<li class="toctree-l3"><a class="reference internal" href="rfft.html">RFFT</a></li>
<li class="toctree-l3"><a class="reference internal" href="roipooling.html">ROIPooling</a></li>
<li class="toctree-l3"><a class="reference internal" href="round.html">Round</a></li>
<li class="toctree-l3"><a class="reference internal" href="rsqrt.html">Rsqrt</a></li>
<li class="toctree-l3"><a class="reference internal" href="rsqrt_grad.html">RsqrtGrad</a></li>
<li class="toctree-l3"><a class="reference internal" href="scale_fusion.html">ScaleFusion</a></li>
<li class="toctree-l3"><a class="reference internal" href="scatter_elements.html">ScatterElements</a></li>
<li class="toctree-l3"><a class="reference internal" href="scatter_nd.html">ScatterNd</a></li>
<li class="toctree-l3"><a class="reference internal" href="scatter_nd_update.html">ScatterNdUpdate</a></li>
<li class="toctree-l3"><a class="reference internal" href="select.html">Select</a></li>
<li class="toctree-l3"><a class="reference internal" href="sgd.html">SGD</a></li>
<li class="toctree-l3"><a class="reference internal" href="shape.html">Shape</a></li>
<li class="toctree-l3"><a class="reference internal" href="sigmoidcrossentropywithlogits.html">SigmoidCrossEntropyWithLogits</a></li>
<li class="toctree-l3"><a class="reference internal" href="sigmoidcrossentropywithlogitsgrad.html">SigmoidCrossEntropyWithLogitsGrad</a></li>
<li class="toctree-l3"><a class="reference internal" href="sin.html">Sin</a></li>
<li class="toctree-l3"><a class="reference internal" href="size.html">Size</a></li>
<li class="toctree-l3"><a class="reference internal" href="skipgram.html">Skipgram</a></li>
<li class="toctree-l3"><a class="reference internal" href="slice.html">SliceFusion</a></li>
<li class="toctree-l3"><a class="reference internal" href="smooth1loss.html">SmoothL1Loss</a></li>
<li class="toctree-l3"><a class="reference internal" href="smoothl1lossgrad.html">Smoothl1lossgrad</a></li>
<li class="toctree-l3"><a class="reference internal" href="softmax.html">Softmax</a></li>
<li class="toctree-l3"><a class="reference internal" href="softmax_cross_entropy_with_logits.html">SoftmaxCrossEntropyWithLogits</a></li>
<li class="toctree-l3"><a class="reference internal" href="spacetobatch.html">SpaceToBatch</a></li>
<li class="toctree-l3"><a class="reference internal" href="spacetobatchnd.html">SpaceToBatchND</a></li>
<li class="toctree-l3"><a class="reference internal" href="spacetodepth.html">SpaceToDepth</a></li>
<li class="toctree-l3"><a class="reference internal" href="sparsefillemptyrows.html">SparseFillEmptyRows</a></li>
<li class="toctree-l3"><a class="reference internal" href="sparsereshape.html">Sparsereshape</a></li>
<li class="toctree-l3"><a class="reference internal" href="sparsesegmentsum.html">Sparsesegmentsum</a></li>
<li class="toctree-l3"><a class="reference internal" href="sparse_softmax_cross_entropy_with_logits.html">SparseSoftmaxCrossEntropyWithLogits</a></li>
<li class="toctree-l3"><a class="reference internal" href="sparsetodense.html">Sparsetodense</a></li>
<li class="toctree-l3"><a class="reference internal" href="splice.html">Splice</a></li>
<li class="toctree-l3"><a class="reference internal" href="split.html">Split</a></li>
<li class="toctree-l3"><a class="reference internal" href="split_with_overlap.html">SplitWithOverlap</a></li>
<li class="toctree-l3"><a class="reference internal" href="sqrt.html">Sqrt</a></li>
<li class="toctree-l3"><a class="reference internal" href="sqrt_grad.html">SqrtGrad</a></li>
<li class="toctree-l3"><a class="reference internal" href="square.html">Square</a></li>
<li class="toctree-l3"><a class="reference internal" href="squaredifference.html">Squaredifference</a></li>
<li class="toctree-l3"><a class="reference internal" href="squeeze.html">Squeeze</a></li>
<li class="toctree-l3"><a class="reference internal" href="stack.html">Stack</a></li>
<li class="toctree-l3"><a class="reference internal" href="stridedslice.html">Stridedslice</a></li>
<li class="toctree-l3"><a class="reference internal" href="stridedslicegrad.html">Stridedslicegrad</a></li>
<li class="toctree-l3"><a class="reference internal" href="sub.html">SubFusion</a></li>
<li class="toctree-l3"><a class="reference internal" href="sub_grad.html">SubGrad</a></li>
<li class="toctree-l3 current"><a class="current reference internal" href="#">Switch</a></li>
<li class="toctree-l3"><a class="reference internal" href="switchlayer.html">Switchlayer</a></li>
<li class="toctree-l3"><a class="reference internal" href="tensorarray.html">Tensorarray</a></li>
<li class="toctree-l3"><a class="reference internal" href="tensorarrayread.html">Tensorarrayread</a></li>
<li class="toctree-l3"><a class="reference internal" href="tensorarraywrite.html">TensorarrayWrite</a></li>
<li class="toctree-l3"><a class="reference internal" href="tensorlist_from_tensor.html">TensorListFromTensor</a></li>
<li class="toctree-l3"><a class="reference internal" href="tensorlistgetitem.html">Tensorlistgetitem</a></li>
<li class="toctree-l3"><a class="reference internal" href="tensorlistreserve.html">Tensorlistreserve</a></li>
<li class="toctree-l3"><a class="reference internal" href="tensorlistsetitem.html">Tensorlistsetitem</a></li>
<li class="toctree-l3"><a class="reference internal" href="tensorliststack.html">Tensorliststack</a></li>
<li class="toctree-l3"><a class="reference internal" href="tensor_scatter_add.html">TensorScatterAdd</a></li>
<li class="toctree-l3"><a class="reference internal" href="tile.html">Tile</a></li>
<li class="toctree-l3"><a class="reference internal" href="topkfusion.html">TopkFusion</a></li>
<li class="toctree-l3"><a class="reference internal" href="transpose.html">Transpose</a></li>
<li class="toctree-l3"><a class="reference internal" href="tril.html">Tril</a></li>
<li class="toctree-l3"><a class="reference internal" href="triu.html">Triu</a></li>
<li class="toctree-l3"><a class="reference internal" href="uniform_real.html">UniformReal</a></li>
<li class="toctree-l3"><a class="reference internal" href="unique.html">Unique</a></li>
<li class="toctree-l3"><a class="reference internal" href="unsortedsegmentsum.html">UnsortedSegmentSum</a></li>
<li class="toctree-l3"><a class="reference internal" href="unstack.html">Unstack</a></li>
<li class="toctree-l3"><a class="reference internal" href="unsqueeze.html">UnSqueeze</a></li>
<li class="toctree-l3"><a class="reference internal" href="where.html">Where</a></li>
<li class="toctree-l3"><a class="reference internal" href="zeroslike.html">ZerosLike</a></li>
<li class="toctree-l3"><a class="reference internal" href="call.html">Call</a></li>
<li class="toctree-l3"><a class="reference internal" href="custom.html">Custom</a></li>
<li class="toctree-l3"><a class="reference internal" href="depend.html">Depend</a></li>
<li class="toctree-l3"><a class="reference internal" href="genop.html">GenOp</a></li>
<li class="toctree-l3"><a class="reference internal" href="partialfusion.html">PartialFusion</a></li>
</ul>
</li>
</ul>
</li>
<li class="toctree-l1"><a class="reference internal" href="../../refdoc/index.html">参考资料</a></li>
</ul>
</div>
</div>
</nav>
<section data-toggle="wy-nav-shift" class="wy-nav-content-wrap"><nav class="wy-nav-top" aria-label="移动版导航菜单" >
<i data-toggle="wy-nav-top" class="fa fa-bars"></i>
<a href="../../index.html">MindSpore Signal+ 使用手册</a>
</nav>
<div class="wy-nav-content">
<div class="rst-content">
<div role="navigation" aria-label="页面导航">
<ul class="wy-breadcrumbs">
<li><a href="../../index.html" class="icon icon-home" aria-label="Home"></a></li>
<li class="breadcrumb-item"><a href="../index.html">算子库支持</a></li>
<li class="breadcrumb-item"><a href="dsplib_index.html">DSP Library C API Reference</a></li>
<li class="breadcrumb-item active">Switch</li>
<li class="wy-breadcrumbs-aside">
<a href="../../_sources/functionlib/dsplib/switch.rst.txt" rel="nofollow"> 查看页面源码</a>
</li>
</ul>
<hr/>
</div>
<div role="main" class="document" itemscope="itemscope" itemtype="http://schema.org/Article">
<div itemprop="articleBody">
<section id="switch">
<h1>Switch<a class="headerlink" href="#switch" title="此标题的永久链接"></a></h1>
<p>根据条件选择输入张量。该算子根据布尔条件 <cite>condition</cite> 的值,选择 <cite>input_x</cite><cite>input_y</cite> 作为输出。该算子不区分数据类型,适用于所有数据类型。</p>
<div class="math notranslate nohighlight">
\[\begin{split}\text{output} = \begin{cases}
\text{input\_x}, &amp; \text{if } \text{condition} = \text{True} \\
\text{input\_y}, &amp; \text{if } \text{condition} = \text{False}
\end{cases}\end{split}\]</div>
<p>该算子不复制数据,只是将输出指针指向选中的输入张量。因此,输出张量共享输入张量的数据指针和元数据。</p>
<dl class="simple">
<dt>输入:</dt><dd><ul class="simple">
<li><p><strong>input_x</strong> - 第一个输入张量TensorC* 类型)。当 <cite>condition</cite> 为 True 时被选中。</p></li>
<li><p><strong>input_y</strong> - 第二个输入张量TensorC* 类型)。当 <cite>condition</cite> 为 False 时被选中。</p></li>
<li><p><strong>condition</strong> - 条件值bool 类型),决定选择哪个输入张量。</p></li>
</ul>
</dd>
<dt>输出:</dt><dd><ul class="simple">
<li><p><strong>output</strong> - 输出张量指针的指针TensorC** 类型),指向选中的输入张量。</p></li>
</ul>
</dd>
<dt>支持平台:</dt><dd><p><code class="docutils literal notranslate"><span class="pre">FT78NE</span></code>
<code class="docutils literal notranslate"><span class="pre">MT7004</span></code></p>
</dd>
</dl>
<div class="admonition note">
<p class="admonition-title">备注</p>
<ul class="simple">
<li><p>该算子不区分数据类型,适用于所有数据类型</p></li>
<li><p>算子不复制数据,输出张量共享输入张量的数据指针</p></li>
<li><p>输出张量的所有元数据(形状、数据类型、格式等)与选中的输入张量相同</p></li>
</ul>
</div>
<p><strong>共享存储版本:</strong></p>
<dl class="c function">
<dt class="sig sig-object c" id="c.switch_s">
<span class="kt"><span class="pre">void</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">switch_s</span></span></span><span class="sig-paren">(</span><span class="n"><span class="pre">TensorC</span></span><span class="w"> </span><span class="p"><span class="pre">*</span></span><span class="n"><span class="pre">input_x</span></span>, <span class="n"><span class="pre">TensorC</span></span><span class="w"> </span><span class="p"><span class="pre">*</span></span><span class="n"><span class="pre">input_y</span></span>, <span class="n"><span class="pre">TensorC</span></span><span class="w"> </span><span class="p"><span class="pre">*</span></span><span class="p"><span class="pre">*</span></span><span class="n"><span class="pre">output</span></span>, <span class="kt"><span class="pre">bool</span></span><span class="w"> </span><span class="n"><span class="pre">condition</span></span><span class="sig-paren">)</span><a class="headerlink" href="#c.switch_s" title="永久链接至目标"></a><br /></dt>
<dd></dd></dl>
<p><strong>C调用示例</strong></p>
<div class="highlight-c notranslate"><div class="highlight"><pre><span></span><span class="linenos"> 1</span><span class="c1">//FT78NE示例</span>
<span class="linenos"> 2</span><span class="cp">#include</span><span class="w"> </span><span class="cpf">&lt;stdio.h&gt;</span>
<span class="linenos"> 3</span><span class="cp">#include</span><span class="w"> </span><span class="cpf">&lt;switch.h&gt;</span>
<span class="linenos"> 4</span>
<span class="linenos"> 5</span><span class="kt">int</span><span class="w"> </span><span class="nf">main</span><span class="p">(</span><span class="kt">int</span><span class="w"> </span><span class="n">argc</span><span class="p">,</span><span class="w"> </span><span class="kt">char</span><span class="o">*</span><span class="w"> </span><span class="n">argv</span><span class="p">[])</span><span class="w"> </span><span class="p">{</span>
<span class="linenos"> 6</span><span class="w"> </span><span class="n">TensorC</span><span class="w"> </span><span class="n">input_x</span><span class="p">;</span>
<span class="linenos"> 7</span><span class="w"> </span><span class="n">TensorC</span><span class="w"> </span><span class="n">input_y</span><span class="p">;</span>
<span class="linenos"> 8</span><span class="w"> </span><span class="n">TensorC</span><span class="o">*</span><span class="w"> </span><span class="n">output</span><span class="p">;</span>
<span class="linenos"> 9</span>
<span class="linenos">10</span><span class="w"> </span><span class="c1">// 初始化 input_x</span>
<span class="linenos">11</span><span class="w"> </span><span class="kt">int</span><span class="w"> </span><span class="n">x_shape</span><span class="p">[</span><span class="mi">3</span><span class="p">]</span><span class="w"> </span><span class="o">=</span><span class="w"> </span><span class="p">{</span><span class="mi">2</span><span class="p">,</span><span class="w"> </span><span class="mi">3</span><span class="p">,</span><span class="w"> </span><span class="mi">4</span><span class="p">};</span>
<span class="linenos">12</span><span class="w"> </span><span class="n">memcpy</span><span class="p">(</span><span class="n">input_x</span><span class="p">.</span><span class="n">shape_</span><span class="p">,</span><span class="w"> </span><span class="n">x_shape</span><span class="p">,</span><span class="w"> </span><span class="mi">3</span><span class="w"> </span><span class="o">*</span><span class="w"> </span><span class="k">sizeof</span><span class="p">(</span><span class="kt">int</span><span class="p">));</span>
<span class="linenos">13</span><span class="w"> </span><span class="n">input_x</span><span class="p">.</span><span class="n">shape_size_</span><span class="w"> </span><span class="o">=</span><span class="w"> </span><span class="mi">3</span><span class="p">;</span>
<span class="linenos">14</span><span class="w"> </span><span class="n">input_x</span><span class="p">.</span><span class="n">data_type_</span><span class="w"> </span><span class="o">=</span><span class="w"> </span><span class="n">kNumberTypeFloat32</span><span class="p">;</span>
<span class="linenos">15</span><span class="w"> </span><span class="n">input_x</span><span class="p">.</span><span class="n">format_</span><span class="w"> </span><span class="o">=</span><span class="w"> </span><span class="n">Format_NCHW</span><span class="p">;</span>
<span class="linenos">16</span><span class="w"> </span><span class="n">input_x</span><span class="p">.</span><span class="n">data_</span><span class="w"> </span><span class="o">=</span><span class="w"> </span><span class="p">(</span><span class="kt">void</span><span class="w"> </span><span class="o">*</span><span class="p">)</span><span class="mh">0xA0000000</span><span class="p">;</span>
<span class="linenos">17</span><span class="w"> </span><span class="n">input_x</span><span class="p">.</span><span class="n">category_</span><span class="w"> </span><span class="o">=</span><span class="w"> </span><span class="mi">0</span><span class="p">;</span><span class="w"> </span><span class="c1">// 非常量</span>
<span class="linenos">18</span><span class="w"> </span><span class="n">input_x</span><span class="p">.</span><span class="n">shape_changed_</span><span class="w"> </span><span class="o">=</span><span class="w"> </span><span class="nb">false</span><span class="p">;</span>
<span class="linenos">19</span>
<span class="linenos">20</span><span class="w"> </span><span class="c1">// 初始化 input_y</span>
<span class="linenos">21</span><span class="w"> </span><span class="kt">int</span><span class="w"> </span><span class="n">y_shape</span><span class="p">[</span><span class="mi">3</span><span class="p">]</span><span class="w"> </span><span class="o">=</span><span class="w"> </span><span class="p">{</span><span class="mi">2</span><span class="p">,</span><span class="w"> </span><span class="mi">3</span><span class="p">,</span><span class="w"> </span><span class="mi">4</span><span class="p">};</span>
<span class="linenos">22</span><span class="w"> </span><span class="n">memcpy</span><span class="p">(</span><span class="n">input_y</span><span class="p">.</span><span class="n">shape_</span><span class="p">,</span><span class="w"> </span><span class="n">y_shape</span><span class="p">,</span><span class="w"> </span><span class="mi">3</span><span class="w"> </span><span class="o">*</span><span class="w"> </span><span class="k">sizeof</span><span class="p">(</span><span class="kt">int</span><span class="p">));</span>
<span class="linenos">23</span><span class="w"> </span><span class="n">input_y</span><span class="p">.</span><span class="n">shape_size_</span><span class="w"> </span><span class="o">=</span><span class="w"> </span><span class="mi">3</span><span class="p">;</span>
<span class="linenos">24</span><span class="w"> </span><span class="n">input_y</span><span class="p">.</span><span class="n">data_type_</span><span class="w"> </span><span class="o">=</span><span class="w"> </span><span class="n">kNumberTypeFloat32</span><span class="p">;</span>
<span class="linenos">25</span><span class="w"> </span><span class="n">input_y</span><span class="p">.</span><span class="n">format_</span><span class="w"> </span><span class="o">=</span><span class="w"> </span><span class="n">Format_NCHW</span><span class="p">;</span>
<span class="linenos">26</span><span class="w"> </span><span class="n">input_y</span><span class="p">.</span><span class="n">data_</span><span class="w"> </span><span class="o">=</span><span class="w"> </span><span class="p">(</span><span class="kt">void</span><span class="w"> </span><span class="o">*</span><span class="p">)</span><span class="mh">0xB0000000</span><span class="p">;</span>
<span class="linenos">27</span><span class="w"> </span><span class="n">input_y</span><span class="p">.</span><span class="n">category_</span><span class="w"> </span><span class="o">=</span><span class="w"> </span><span class="mi">0</span><span class="p">;</span>
<span class="linenos">28</span><span class="w"> </span><span class="n">input_y</span><span class="p">.</span><span class="n">shape_changed_</span><span class="w"> </span><span class="o">=</span><span class="w"> </span><span class="nb">false</span><span class="p">;</span>
<span class="linenos">29</span>
<span class="linenos">30</span><span class="w"> </span><span class="kt">bool</span><span class="w"> </span><span class="n">condition</span><span class="w"> </span><span class="o">=</span><span class="w"> </span><span class="nb">true</span><span class="p">;</span><span class="w"> </span><span class="c1">// 选择 input_x</span>
<span class="linenos">31</span>
<span class="hll"><span class="linenos">32</span><span class="w"> </span><span class="n">switch_s</span><span class="p">(</span><span class="o">&amp;</span><span class="n">input_x</span><span class="p">,</span><span class="w"> </span><span class="o">&amp;</span><span class="n">input_y</span><span class="p">,</span><span class="w"> </span><span class="o">&amp;</span><span class="n">output</span><span class="p">,</span><span class="w"> </span><span class="n">condition</span><span class="p">);</span>
</span><span class="linenos">33</span>
<span class="linenos">34</span><span class="w"> </span><span class="k">return</span><span class="w"> </span><span class="mi">0</span><span class="p">;</span>
<span class="linenos">35</span><span class="p">}</span>
</pre></div>
</div>
<p><strong>私有存储版本:</strong></p>
<dl class="c function">
<dt class="sig sig-object c" id="c.switch_p">
<span class="kt"><span class="pre">void</span></span><span class="w"> </span><span class="sig-name descname"><span class="n"><span class="pre">switch_p</span></span></span><span class="sig-paren">(</span><span class="n"><span class="pre">TensorC</span></span><span class="w"> </span><span class="p"><span class="pre">*</span></span><span class="n"><span class="pre">input_x</span></span>, <span class="n"><span class="pre">TensorC</span></span><span class="w"> </span><span class="p"><span class="pre">*</span></span><span class="n"><span class="pre">input_y</span></span>, <span class="n"><span class="pre">TensorC</span></span><span class="w"> </span><span class="p"><span class="pre">*</span></span><span class="p"><span class="pre">*</span></span><span class="n"><span class="pre">output</span></span>, <span class="kt"><span class="pre">bool</span></span><span class="w"> </span><span class="n"><span class="pre">condition</span></span><span class="sig-paren">)</span><a class="headerlink" href="#c.switch_p" title="永久链接至目标"></a><br /></dt>
<dd></dd></dl>
<p><strong>C调用示例</strong></p>
<div class="highlight-c notranslate"><div class="highlight"><pre><span></span><span class="linenos"> 1</span><span class="c1">//FT78NE示例</span>
<span class="linenos"> 2</span><span class="cp">#include</span><span class="w"> </span><span class="cpf">&lt;stdio.h&gt;</span>
<span class="linenos"> 3</span><span class="cp">#include</span><span class="w"> </span><span class="cpf">&lt;switch.h&gt;</span>
<span class="linenos"> 4</span>
<span class="linenos"> 5</span><span class="kt">int</span><span class="w"> </span><span class="nf">main</span><span class="p">(</span><span class="kt">int</span><span class="w"> </span><span class="n">argc</span><span class="p">,</span><span class="w"> </span><span class="kt">char</span><span class="o">*</span><span class="w"> </span><span class="n">argv</span><span class="p">[])</span><span class="w"> </span><span class="p">{</span>
<span class="linenos"> 6</span><span class="w"> </span><span class="n">TensorC</span><span class="w"> </span><span class="n">input_x</span><span class="p">;</span>
<span class="linenos"> 7</span><span class="w"> </span><span class="n">TensorC</span><span class="w"> </span><span class="n">input_y</span><span class="p">;</span>
<span class="linenos"> 8</span><span class="w"> </span><span class="n">TensorC</span><span class="o">*</span><span class="w"> </span><span class="n">output</span><span class="p">;</span>
<span class="linenos"> 9</span>
<span class="linenos">10</span><span class="w"> </span><span class="c1">// 初始化 input_x</span>
<span class="linenos">11</span><span class="w"> </span><span class="kt">int</span><span class="w"> </span><span class="n">x_shape</span><span class="p">[</span><span class="mi">3</span><span class="p">]</span><span class="w"> </span><span class="o">=</span><span class="w"> </span><span class="p">{</span><span class="mi">2</span><span class="p">,</span><span class="w"> </span><span class="mi">3</span><span class="p">,</span><span class="w"> </span><span class="mi">4</span><span class="p">};</span>
<span class="linenos">12</span><span class="w"> </span><span class="n">memcpy</span><span class="p">(</span><span class="n">input_x</span><span class="p">.</span><span class="n">shape_</span><span class="p">,</span><span class="w"> </span><span class="n">x_shape</span><span class="p">,</span><span class="w"> </span><span class="mi">3</span><span class="w"> </span><span class="o">*</span><span class="w"> </span><span class="k">sizeof</span><span class="p">(</span><span class="kt">int</span><span class="p">));</span>
<span class="linenos">13</span><span class="w"> </span><span class="n">input_x</span><span class="p">.</span><span class="n">shape_size_</span><span class="w"> </span><span class="o">=</span><span class="w"> </span><span class="mi">3</span><span class="p">;</span>
<span class="linenos">14</span><span class="w"> </span><span class="n">input_x</span><span class="p">.</span><span class="n">data_type_</span><span class="w"> </span><span class="o">=</span><span class="w"> </span><span class="n">kNumberTypeFloat32</span><span class="p">;</span>
<span class="linenos">15</span><span class="w"> </span><span class="n">input_x</span><span class="p">.</span><span class="n">format_</span><span class="w"> </span><span class="o">=</span><span class="w"> </span><span class="n">Format_NCHW</span><span class="p">;</span>
<span class="linenos">16</span><span class="w"> </span><span class="n">input_x</span><span class="p">.</span><span class="n">data_</span><span class="w"> </span><span class="o">=</span><span class="w"> </span><span class="p">(</span><span class="kt">void</span><span class="w"> </span><span class="o">*</span><span class="p">)</span><span class="mh">0x10000000</span><span class="p">;</span>
<span class="linenos">17</span><span class="w"> </span><span class="n">input_x</span><span class="p">.</span><span class="n">category_</span><span class="w"> </span><span class="o">=</span><span class="w"> </span><span class="mi">0</span><span class="p">;</span><span class="w"> </span><span class="c1">// 非常量</span>
<span class="linenos">18</span><span class="w"> </span><span class="n">input_x</span><span class="p">.</span><span class="n">shape_changed_</span><span class="w"> </span><span class="o">=</span><span class="w"> </span><span class="nb">false</span><span class="p">;</span>
<span class="linenos">19</span>
<span class="linenos">20</span><span class="w"> </span><span class="c1">// 初始化 input_y</span>
<span class="linenos">21</span><span class="w"> </span><span class="kt">int</span><span class="w"> </span><span class="n">y_shape</span><span class="p">[</span><span class="mi">3</span><span class="p">]</span><span class="w"> </span><span class="o">=</span><span class="w"> </span><span class="p">{</span><span class="mi">2</span><span class="p">,</span><span class="w"> </span><span class="mi">3</span><span class="p">,</span><span class="w"> </span><span class="mi">4</span><span class="p">};</span>
<span class="linenos">22</span><span class="w"> </span><span class="n">memcpy</span><span class="p">(</span><span class="n">input_y</span><span class="p">.</span><span class="n">shape_</span><span class="p">,</span><span class="w"> </span><span class="n">y_shape</span><span class="p">,</span><span class="w"> </span><span class="mi">3</span><span class="w"> </span><span class="o">*</span><span class="w"> </span><span class="k">sizeof</span><span class="p">(</span><span class="kt">int</span><span class="p">));</span>
<span class="linenos">23</span><span class="w"> </span><span class="n">input_y</span><span class="p">.</span><span class="n">shape_size_</span><span class="w"> </span><span class="o">=</span><span class="w"> </span><span class="mi">3</span><span class="p">;</span>
<span class="linenos">24</span><span class="w"> </span><span class="n">input_y</span><span class="p">.</span><span class="n">data_type_</span><span class="w"> </span><span class="o">=</span><span class="w"> </span><span class="n">kNumberTypeFloat32</span><span class="p">;</span>
<span class="linenos">25</span><span class="w"> </span><span class="n">input_y</span><span class="p">.</span><span class="n">format_</span><span class="w"> </span><span class="o">=</span><span class="w"> </span><span class="n">Format_NCHW</span><span class="p">;</span>
<span class="linenos">26</span><span class="w"> </span><span class="n">input_y</span><span class="p">.</span><span class="n">data_</span><span class="w"> </span><span class="o">=</span><span class="w"> </span><span class="p">(</span><span class="kt">void</span><span class="w"> </span><span class="o">*</span><span class="p">)</span><span class="mh">0x10001000</span><span class="p">;</span>
<span class="linenos">27</span><span class="w"> </span><span class="n">input_y</span><span class="p">.</span><span class="n">category_</span><span class="w"> </span><span class="o">=</span><span class="w"> </span><span class="mi">0</span><span class="p">;</span>
<span class="linenos">28</span><span class="w"> </span><span class="n">input_y</span><span class="p">.</span><span class="n">shape_changed_</span><span class="w"> </span><span class="o">=</span><span class="w"> </span><span class="nb">false</span><span class="p">;</span>
<span class="linenos">29</span>
<span class="linenos">30</span><span class="w"> </span><span class="kt">bool</span><span class="w"> </span><span class="n">condition</span><span class="w"> </span><span class="o">=</span><span class="w"> </span><span class="nb">true</span><span class="p">;</span><span class="w"> </span><span class="c1">// 选择 input_x</span>
<span class="linenos">31</span>
<span class="hll"><span class="linenos">32</span><span class="w"> </span><span class="n">switch_p</span><span class="p">(</span><span class="o">&amp;</span><span class="n">input_x</span><span class="p">,</span><span class="w"> </span><span class="o">&amp;</span><span class="n">input_y</span><span class="p">,</span><span class="w"> </span><span class="o">&amp;</span><span class="n">output</span><span class="p">,</span><span class="w"> </span><span class="n">condition</span><span class="p">);</span>
</span><span class="linenos">33</span>
<span class="linenos">34</span><span class="w"> </span><span class="k">return</span><span class="w"> </span><span class="mi">0</span><span class="p">;</span>
<span class="linenos">35</span><span class="p">}</span>
</pre></div>
</div>
</section>
</div>
</div>
<div class="pageviews-container">
访问量:<span id="busuanzi_page_uv"></span>
</div>
<footer><div class="rst-footer-buttons" role="navigation" aria-label="页脚">
<a href="sub_grad.html" class="btn btn-neutral float-left" title="SubGrad" accesskey="p" rel="prev"><span class="fa fa-arrow-circle-left" aria-hidden="true"></span> 上一页</a>
<a href="switchlayer.html" class="btn btn-neutral float-right" title="Switchlayer" accesskey="n" rel="next">下一页 <span class="fa fa-arrow-circle-right" aria-hidden="true"></span></a>
</div>
<hr/>
<div role="contentinfo">
<p>&#169; 版权所有 2025 - 2026, NUDT-674。</p>
</div>
利用 <a href="https://www.sphinx-doc.org/">Sphinx</a> 构建,使用的
<a href="https://github.com/readthedocs/sphinx_rtd_theme">主题</a>
<a href="https://readthedocs.org">Read the Docs</a> 开发.
</footer>
</div>
</div>
</section>
</div>
<script>
jQuery(function () {
SphinxRtdTheme.Navigation.enable(true);
});
</script>
<script src="//cdn.busuanzi.cc/busuanzi/3.6.9/busuanzi.min.js" defer></script>
</body>
</html>