PyTorch与MindSpore API映射表

由社区提供的PyTorch APIs和MindSpore APIs之间的映射,可能在参数、输入、输出、逻辑功能和特定场景等方面存在差异,可详见各API描述或已提供的差异对比。

也欢迎更多的MindSpore开发者参与完善映射内容。

因为框架机制原因,MindSpore不提供PyTorch的以下参数:

参数名 功能
out 表示输出的Tensor
layout 表示内存分布策略
device 表示Tensor存放位置
requires_grad 表示是否更新梯度
generator 表示伪随机数生成器
pin_memeory 表示是否使用锁页内存
memory_format 表示Tensor的内存格式
stable 表示是否稳定排序
inplace 表示在不更改变量内存地址的情况下,直接修改变量的值
sparse_grad 表示是否对梯度稀疏化
size_average PyTorch废弃参数
reduce PyTorch废弃参数

PyTorch 1.8.1

torch

PyTorch 1.8.1 APIs MindSpore APIs 说明
torch.abs mindspore.ops.abs 一致
torch.absolute mindspore.ops.absolute 一致
torch.acos mindspore.ops.acos 一致
torch.acosh mindspore.ops.acosh 一致
torch.add mindspore.ops.add 一致
torch.addbmm mindspore.ops.addbmm 功能一致,参数名不同
torch.addcdiv mindspore.ops.addcdiv 一致
torch.addcmul mindspore.ops.addcmul 一致
torch.addmm mindspore.ops.addmm 功能一致,参数名不同
torch.addmv mindspore.ops.addmv 功能一致,参数名不同
torch.addr mindspore.ops.addr 功能一致,参数名不同
torch.all mindspore.ops.all 差异对比
torch.amax mindspore.ops.amax 差异对比
torch.amin mindspore.ops.amin 差异对比
torch.angle mindspore.ops.angle 一致
torch.any mindspore.ops.any 差异对比
torch.arange mindspore.ops.arange 一致
torch.arccos mindspore.ops.arccos 一致
torch.arccosh mindspore.ops.arccosh 一致
torch.arcsin mindspore.ops.arcsin 功能一致,参数名不同
torch.arcsinh mindspore.ops.arcsinh 功能一致,参数名不同
torch.arctan mindspore.ops.arctan 一致
torch.arctanh mindspore.ops.arctanh 功能一致,参数名不同
torch.argmax mindspore.ops.argmax 一致
torch.argmin mindspore.ops.argmin 差异对比
torch.argsort mindspore.ops.argsort 一致
torch.asin mindspore.ops.asin 一致
torch.asinh mindspore.ops.asinh 一致
torch.atan mindspore.ops.atan 一致
torch.atan2 mindspore.ops.atan2 一致
torch.atanh mindspore.ops.atanh 一致
torch.atleast_1d mindspore.ops.atleast_1d 功能一致,参数名不同
torch.atleast_2d mindspore.ops.atleast_2d 功能一致,参数名不同
torch.atleast_3d mindspore.ops.atleast_3d 功能一致,参数名不同
torch.baddbmm mindspore.ops.baddbmm 一致
torch.bartlett_window mindspore.ops.bartlett_window 差异对比
torch.bernoulli mindspore.ops.bernoulli 差异对比
torch.bincount mindspore.ops.bincount 一致
torch.bitwise_and mindspore.ops.bitwise_and 一致
torch.bitwise_or mindspore.ops.bitwise_or 一致
torch.bitwise_xor mindspore.ops.bitwise_xor 一致
torch.blackman_window mindspore.ops.blackman_window 差异对比
torch.block_diag mindspore.ops.block_diag 功能一致,参数名不同
torch.bmm mindspore.ops.bmm 功能一致,参数名不同
torch.broadcast_to mindspore.ops.broadcast_to 差异对比
torch.cat mindspore.ops.cat 差异对比
torch.cdist mindspore.ops.cdist 差异对比
torch.ceil mindspore.ops.ceil 一致
torch.cholesky mindspore.ops.cholesky 功能一致,参数名不同
torch.chunk mindspore.ops.chunk 一致
torch.clamp mindspore.ops.clamp 一致
torch.clip mindspore.ops.clip 差异对比
torch.column_stack mindspore.ops.column_stack 一致
torch.combinations mindspore.ops.combinations 功能一致,参数名不同
torch.conj mindspore.ops.conj 一致
torch.copysign mindspore.ops.copysign 功能一致,参数名不同
torch.cos mindspore.ops.cos 一致
torch.cosh mindspore.ops.cosh 一致
torch.count_nonzero mindspore.ops.count_nonzero 一致
torch.cross mindspore.ops.cross 一致
torch.cummax mindspore.ops.cummax 一致
torch.cummin mindspore.ops.cummin 一致
torch.cumprod mindspore.ops.cumprod 一致
torch.cumsum mindspore.ops.cumsum 差异对比
torch.deg2rad mindspore.ops.deg2rad 功能一致,参数名不同
torch.diag mindspore.ops.diag 差异对比
torch.diag_embed mindspore.ops.diag_embed 一致
torch.diagflat mindspore.ops.diagflat 一致
torch.diagonal mindspore.ops.diagonal 一致
torch.diff mindspore.ops.diff 功能一致,参数名不同
torch.digamma mindspore.ops.digamma 一致
torch.dist mindspore.ops.dist 一致
torch.div mindspore.ops.div 差异对比
torch.divide mindspore.ops.divide 一致
torch.dot mindspore.ops.tensor_dot 差异对比
torch.dstack mindspore.ops.dstack 功能一致,参数名不同
torch.empty mindspore.numpy.empty 差异对比
torch.empty_like mindspore.numpy.empty_like 差异对比
torch.eig mindspore.ops.eig 差异对比
torch.einsum mindspore.ops.einsum 一致
torch.eq mindspore.ops.equal 差异对比
torch.equal mindspore.ops.equal 一致
torch.erf mindspore.ops.erf 一致
torch.erfc mindspore.ops.erfc 一致
torch.erfinv mindspore.ops.erfinv 一致
torch.exp mindspore.ops.exp 一致
torch.exp2 mindspore.ops.exp2 一致
torch.expm1 mindspore.ops.expm1 一致
torch.eye mindspore.ops.eye 一致
torch.flatten mindspore.ops.flatten 差异对比
torch.float_power mindspore.ops.float_power 差异对比
torch.flip mindspore.ops.flip 功能一致,参数名不同
torch.fliplr mindspore.ops.fliplr 功能一致,参数名不同
torch.flipud mindspore.ops.flipud 功能一致,参数名不同
torch.floor mindspore.ops.floor 一致
torch.fmax mindspore.ops.fmax 一致
torch.fmod mindspore.ops.fmod 一致
torch.frac mindspore.ops.frac 功能一致,参数名不同
torch.full mindspore.ops.full 差异对比
torch.full_like mindspore.ops.full_like 一致
torch.gather mindspore.ops.gather_elements 一致
torch.gcd mindspore.ops.gcd 一致
torch.ge mindspore.ops.ge 功能一致,参数名不同
torch.geqrf mindspore.ops.geqrf 一致
torch.ger mindspore.ops.ger 差异对比
torch.greater mindspore.ops.greater 功能一致,参数名不同
torch.greater_equal mindspore.ops.greater_equal 功能一致,参数名不同
torch.gt mindspore.ops.gt 功能一致,参数名不同
torch.hann_window mindspore.numpy.hanning 差异对比
torch.hann_window mindspore.ops.hann_window 一致
torch.heaviside mindspore.ops.heaviside 一致
torch.hstack mindspore.ops.hstack 一致
torch.histc mindspore.ops.histc 一致
torch.hypot mindspore.ops.hypot 一致
torch.hamming_window mindspore.ops.hamming_window 一致
torch.i0 mindspore.ops.i0 一致
torch.igamma mindspore.ops.igamma 一致
torch.igammac mindspore.ops.igammac 一致
torch.imag mindspore.ops.imag 一致
torch.index_select mindspore.ops.index_select 功能一致,参数名不同
torch.inner mindspore.ops.inner 功能一致,参数名不同
torch.inverse mindspore.ops.inverse 一致
torch.isclose mindspore.ops.isclose 功能一致,参数名不同
torch.isfinite mindspore.ops.isfinite 一致
torch.isinf mindspore.ops.isinf 一致
torch.isnan mindspore.ops.isnan 一致
torch.isneginf mindspore.ops.isneginf 一致
torch.isposinf mindspore.ops.isposinf 一致
torch.isreal mindspore.ops.isreal 一致
torch.is_complex mindspore.ops.is_complex 功能一致,参数名不同
torch.is_floating_point mindspore.ops.is_floating_point 功能一致,参数名不同
torch.is_tensor mindspore.ops.is_tensor 一致
torch.kaiser_window mindspore.ops.kaiser_window 一致
torch.kron mindspore.ops.kron 功能一致,参数名不同
torch.lcm mindspore.ops.lcm 一致
torch.ldexp mindspore.ops.ldexp 功能一致,参数名不同
torch.le mindspore.ops.le 功能一致,参数名不同
torch.lerp mindspore.ops.lerp 功能一致,参数名不同
torch.less mindspore.ops.less 功能一致,参数名不同
torch.less_equal mindspore.ops.less_equal 功能一致,参数名不同
torch.linalg.norm mindspore.ops.norm 一致
torch.linalg.qr mindspore.ops.qr 功能一致,数据类型不同
torch.linspace mindspore.ops.linspace 一致
torch.log mindspore.ops.log 一致
torch.log2 mindspore.ops.log2 一致
torch.log10 mindspore.ops.log10 一致
torch.logaddexp mindspore.ops.logaddexp 一致
torch.logaddexp2 mindspore.ops.logaddexp2 一致
torch.log1p mindspore.ops.log1p 功能一致,参数名不同
torch.logical_and mindspore.ops.logical_and 功能一致,参数名不同
torch.logical_not mindspore.ops.logical_not 功能一致,参数名不同
torch.logical_or mindspore.ops.logical_or 功能一致,参数名不同
torch.logical_xor mindspore.ops.logical_xor 一致
torch.logit mindspore.ops.logit 功能一致,参数名不同
torch.logspace mindspore.ops.logspace 一致
torch.logsumexp mindspore.ops.logsumexp 一致
torch.lt mindspore.ops.lt 一致
torch.masked_select mindspore.ops.masked_select 一致
torch.matmul mindspore.ops.matmul 一致
torch.max mindspore.ops.max 差异对比
torch.maximum mindspore.ops.maximum 功能一致,参数名不同
torch.mean mindspore.ops.mean 功能一致,参数名不同
torch.median mindspore.ops.median 差异对比
torch.meshgrid mindspore.ops.meshgrid 差异对比
torch.mm mindspore.ops.mm 功能一致,参数名不同
torch.mul mindspore.ops.mul 功能一致,参数名不同
torch.min mindspore.ops.min 差异对比
torch.minimum mindspore.ops.minimum 功能一致,参数名不同
torch.msort mindspore.ops.msort 功能一致,参数名不同
torch.moveaxis mindspore.ops.moveaxis 一致
torch.movedim mindspore.ops.movedim 一致
torch.multinomial mindspore.ops.multinomial 差异对比
torch.multiply mindspore.ops.multiply 一致
torch.mv mindspore.ops.mv 功能一致,参数名不同
torch.mvlgamma mindspore.ops.mvlgamma 一致
torch.nan_to_num mindspore.ops.nan_to_num 功能一致,MindSpore暂时只支持float16与float32输入
torch.nansum mindspore.ops.nansum 一致
torch.narrow mindspore.ops.narrow 功能一致
torch.ne mindspore.ops.ne 功能一致,参数名不同
torch.neg mindspore.ops.neg 功能一致,参数名不同
torch.negative mindspore.ops.negative 一致
torch.nextafter mindspore.ops.nextafter 一致
torch.nonzero mindspore.ops.nonzero 功能一致,参数名不同
torch.normal mindspore.ops.normal 差异对比
torch.not_equal mindspore.ops.not_equal 一致
torch.numel mindspore.ops.numel 一致
torch.ones mindspore.ops.ones 差异对比
torch.ones_like mindspore.ops.ones_like 一致
torch.orgqr mindspore.ops.orgqr 功能一致,参数名不同
torch.outer mindspore.ops.outer 功能一致,参数名不同
torch.poisson mindspore.ops.random_poisson 差异对比
torch.polar mindspore.ops.polar 一致
torch.polygamma mindspore.ops.polygamma 一致
torch.pow mindspore.ops.pow 功能一致,参数名不同
torch.prod mindspore.ops.prod 差异对比
torch.rad2deg mindspore.ops.rad2deg 功能一致,参数名不同
torch.rand mindspore.ops.rand 一致
torch.rand_like mindspore.ops.rand_like 一致
torch.randn mindspore.ops.randn 一致
torch.randn_like mindspore.ops.randn_like 一致
torch.randint mindspore.ops.randint 功能一致,默认值不同
torch.randint_like mindspore.ops.randint_like 功能一致,默认值不同
torch.randperm mindspore.ops.randperm 功能一致,参数名不同
torch.range mindspore.ops.range 差异对比
torch.ravel mindspore.ops.ravel 一致
torch.real mindspore.ops.real 功能一致,参数名不同
torch.reciprocal mindspore.ops.reciprocal 功能一致,参数名不同
torch.remainder mindspore.ops.remainder 一致
torch.renorm mindspore.ops.renorm 差异对比
torch.repeat_interleave mindspore.ops.repeat_interleave 功能一致,参数名不同
torch.reshape mindspore.ops.reshape 一致
torch.rot90 mindspore.ops.rot90 一致
torch.round mindspore.ops.round 功能一致,参数名不同
torch.rsqrt mindspore.ops.rsqrt 功能一致,参数名不同
torch.searchsorted mindspore.ops.searchsorted 一致
torch.scatter mindspore.ops.scatter 差异对比
torch.scatter_add mindspore.ops.tensor_scatter_elements 差异对比
torch.sgn mindspore.ops.sgn 功能一致,参数名不同
torch.sigmoid mindspore.ops.sigmoid 功能一致,参数名不同
torch.sign mindspore.ops.sign 功能一致,参数名不同
torch.signbit mindspore.ops.signbit 功能一致,参数名不同
torch.sin mindspore.ops.sin 一致
torch.sinc mindspore.ops.sinc 一致
torch.sinh mindspore.ops.sinh 功能一致,参数名不同
torch.slogdet mindspore.ops.slogdet 一致
torch.sort mindspore.ops.sort 功能一致,参数名不同
torch.split mindspore.ops.split 功能一致,参数名不同
torch.stack mindspore.ops.stack 一致
torch.squeeze mindspore.ops.squeeze 功能一致,参数名不一致
torch.std mindspore.ops.std 差异对比
torch.std_mean mindspore.ops.std_mean 差异对比
torch.sqrt mindspore.ops.sqrt 功能一致,参数名不同
torch.square mindspore.ops.square 功能一致,参数名不同
torch.sub mindspore.ops.subtract 功能一致,参数名不同
torch.subtract mindspore.ops.subtract 功能一致,参数名不同
torch.sum mindspore.ops.sum 功能一致,参数名不同
torch.svd mindspore.ops.svd 差异对比
torch.swapaxes mindspore.ops.swapaxes 一致
torch.swapdims mindspore.ops.swapdims 一致
torch.t mindspore.ops.t 一致
torch.tan mindspore.ops.tan 功能一致,参数名不同
torch.tanh mindspore.ops.tanh 功能一致,参数名不同
torch.tensor_split mindspore.ops.tensor_split 一致
torch.tile mindspore.ops.tile 功能一致,参数名不同
torch.topk mindspore.ops.topk 功能一致,参数名不同
torch.trace mindspore.ops.trace 一致
torch.transpose mindspore.ops.swapaxes 一致
torch.trapz mindspore.ops.trapz 一致
torch.tril mindspore.ops.tril 一致
torch.tril_indices mindspore.ops.tril_indices 一致
torch.triu_indices mindspore.ops.triu_indices 一致
torch.true_divide mindspore.ops.true_divide 一致
torch.trunc mindspore.ops.trunc 一致
torch.unbind mindspore.ops.unbind 一致
torch.unique mindspore.ops.unique 差异对比
torch.unique_consecutive mindspore.ops.unique_consecutive 一致
torch.unsqueeze mindspore.ops.unsqueeze 一致
torch.var mindspore.ops.var 差异对比
torch.var_mean mindspore.ops.var_mean 差异对比
torch.view_as_real mindspore.ops.view_as_real 一致
torch.vstack mindspore.ops.vstack 功能一致,参数名不同
torch.where mindspore.ops.where 一致
torch.xlogy mindspore.ops.xlogy 功能一致,参数名不同
torch.zeros mindspore.ops.zeros 差异对比
torch.zeros_like mindspore.ops.zeros_like 一致

torch.distributions

PyTorch 1.8.1 APIs MindSpore APIs 说明
torch.distributions.laplace.Laplace mindspore.ops.standard_laplace 差异对比

torch.nn

PyTorch 1.8.1 APIs MindSpore APIs 说明
torch.nn.AdaptiveAvgPool1d mindspore.nn.AdaptiveAvgPool1d 一致
torch.nn.AdaptiveAvgPool2d mindspore.nn.AdaptiveAvgPool2d 一致
torch.nn.AdaptiveAvgPool3d mindspore.nn.AdaptiveAvgPool3d 一致
torch.nn.AdaptiveMaxPool1d mindspore.nn.AdaptiveMaxPool1d 差异对比
torch.nn.AdaptiveMaxPool2d mindspore.nn.AdaptiveMaxPool2d 一致
torch.nn.AdaptiveMaxPool3d mindspore.nn.AdaptiveMaxPool3d 一致
torch.nn.AvgPool1d mindspore.nn.AvgPool1d 差异对比
torch.nn.AvgPool2d mindspore.nn.AvgPool2d 差异对比
torch.nn.AvgPool3d mindspore.nn.AvgPool3d 差异对比
torch.nn.BCEWithLogitsLoss mindspore.nn.BCEWithLogitsLoss 差异对比
torch.nn.BatchNorm1d mindspore.nn.BatchNorm1d 差异对比
torch.nn.BatchNorm2d mindspore.nn.BatchNorm2d 差异对比
torch.nn.BatchNorm3d mindspore.nn.BatchNorm3d 差异对比
torch.nn.Bilinear mindspore.nn.BiDense 差异对比
torch.nn.CeLU mindspore.nn.CELU 一致
torch.nn.ChannelShuffle mindspore.nn.ChannelShuffle 一致
torch.nn.CTCLoss mindspore.nn.CTCLoss 一致
torch.nn.ConstantPad1d mindspore.nn.ConstantPad1d 一致
torch.nn.ConstantPad2d mindspore.nn.ConstantPad2d 一致
torch.nn.ConstantPad3d mindspore.nn.ConstantPad3d 一致
torch.nn.Conv1d mindspore.nn.Conv1d 差异对比
torch.nn.Conv2d mindspore.nn.Conv2d 差异对比
torch.nn.Conv3d mindspore.nn.Conv3d 差异对比
torch.nn.ConvTranspose1d mindspore.nn.Conv1dTranspose 差异对比
torch.nn.ConvTranspose2d mindspore.nn.Conv2dTranspose 差异对比
torch.nn.ConvTranspose3d mindspore.nn.Conv3dTranspose 差异对比
torch.nn.CosineEmbeddingLoss mindspore.nn.CosineEmbeddingLoss 差异对比
torch.nn.CrossEntropyLoss mindspore.nn.CrossEntropyLoss 差异对比
torch.nn.Dropout mindspore.nn.Dropout 差异对比
torch.nn.Dropout2d mindspore.nn.Dropout2d 一致
torch.nn.Dropout3d mindspore.nn.Dropout3d 一致
torch.nn.ELU mindspore.nn.ELU 一致
torch.nn.GaussianNLLLoss mindspore.nn.GaussianNLLLoss 差异对比
torch.nn.GELU mindspore.nn.GELU 差异对比
torch.nn.GRU mindspore.nn.GRU 差异对比
torch.nn.GRUCell mindspore.nn.GRUCell 差异对比
torch.nn.GroupNorm mindspore.nn.GroupNorm 差异对比
torch.nn.Hardshrink mindspore.nn.HShrink 差异对比
torch.nn.Hardsigmoid mindspore.nn.Hsigmoid 一致
torch.nn.Hardswish mindspore.nn.Hswish 一致
torch.nn.Hardtanh mindspore.nn.Hardtanh 一致
torch.nn.HingeEmbeddingLoss mindspore.nn.HingeEmbeddingLoss 一致
torch.nn.Identity mindspore.nn.Identity 功能一致,参数名不同
torch.nn.init.uniform_ mindspore.ops.uniform 差异对比
torch.nn.InstanceNorm1d mindspore.nn.InstanceNorm1d 差异对比
torch.nn.InstanceNorm2d mindspore.nn.InstanceNorm2d 差异对比
torch.nn.InstanceNorm3d mindspore.nn.InstanceNorm3d 差异对比
torch.nn.KLDivLoss mindspore.nn.KLDivLoss 一致
torch.nn.L1Loss mindspore.nn.L1Loss 差异对比
torch.nn.LayerNorm mindspore.nn.LayerNorm 差异对比
torch.nn.LeakyReLU mindspore.nn.LeakyReLU 差异对比
torch.nn.LPPool1d mindspore.nn.LPPool1d 一致
torch.nn.LPPool2d mindspore.nn.LPPool2d 一致
torch.nn.LSTM mindspore.nn.LSTM 差异对比
torch.nn.LSTMCell mindspore.nn.LSTMCell 差异对比
torch.nn.Linear mindspore.nn.Dense 差异对比
torch.nn.LocalResponseNorm mindspore.ops.LRN 差异对比
torch.nn.LogSigmoid mindspore.nn.LogSigmoid 一致
torch.nn.MSELoss mindspore.nn.MSELoss 差异对比
torch.nn.MarginRankingLoss mindspore.nn.MarginRankingLoss 一致
torch.nn.MaxPool1d mindspore.nn.MaxPool1d 差异对比
torch.nn.MaxPool2d mindspore.nn.MaxPool2d 差异对比
torch.nn.MaxPool3d mindspore.nn.MaxPool3d 差异对比
torch.nn.MaxUnpool1d mindspore.nn.MaxUnpool1d 一致
torch.nn.MaxUnpool2d mindspore.nn.MaxUnpool2d 一致
torch.nn.MaxUnpool3d mindspore.nn.MaxUnpool3d 一致
torch.nn.MultiheadAttention mindspore.nn.MultiheadAttention 一致
torch.nn.MultiLabelSoftMarginLoss mindspore.nn.MultiLabelSoftMarginLoss 一致
torch.nn.MultiMarginLoss mindspore.nn.MultiMarginLoss 一致
torch.nn.NLLLoss mindspore.nn.NLLLoss 差异对比
torch.nn.PReLU mindspore.nn.PReLU 差异对比
torch.nn.PixelShuffle mindspore.nn.PixelShuffle 一致
torch.nn.PixelUnshuffle mindspore.nn.PixelUnshuffle 一致
torch.nn.PoissonNLLLoss mindspore.nn.PoissonNLLLoss 一致
torch.nn.ReflectionPad1d mindspore.nn.ReflectionPad1d 一致
torch.nn.ReflectionPad2d mindspore.nn.ReflectionPad2d 一致
torch.nn.ReLU mindspore.nn.ReLU 差异对比
torch.nn.ReLU6 mindspore.nn.ReLU6 一致
torch.nn.ReplicationPad1d mindspore.nn.ReplicationPad1d 一致
torch.nn.ReplicationPad2d mindspore.nn.ReplicationPad2d 一致
torch.nn.ReplicationPad3d mindspore.nn.ReplicationPad3d 一致
torch.nn.RNN mindspore.nn.RNN 差异对比
torch.nn.RNNCell mindspore.nn.RNNCell 差异对比
torch.nn.RReLU mindspore.nn.RReLU 一致
torch.nn.SeLU mindspore.nn.SeLU 一致
torch.nn.Sequential mindspore.nn.SequentialCell 差异对比
torch.nn.Sigmoid mindspore.nn.Sigmoid 差异对比
torch.nn.SiLU mindspore.nn.SiLU 一致
torch.nn.SmoothL1Loss mindspore.nn.SmoothL1Loss 差异对比
torch.nn.SoftMarginLoss mindspore.nn.SoftMarginLoss 一致
torch.nn.Softmax mindspore.nn.Softmax 差异对比
torch.nn.Softmin mindspore.nn.Softmin 差异对比
torch.nn.Softmax2d mindspore.nn.Softmax2d 一致
torch.nn.Softshrink mindspore.nn.SoftShrink 差异对比
torch.nn.Softsign mindspore.nn.Softsign 一致
torch.nn.Tanh mindspore.nn.Tanh 差异对比
torch.nn.Tanhshrink mindspore.nn.Tanhshrink 一致
torch.nn.Threshold mindspore.nn.Threshold 一致
torch.nn.Transformer mindspore.nn.Transformer 一致
torch.nn.TransformerDecoder mindspore.nn.TransformerDecoder 一致
torch.nn.TransformerEncoder mindspore.nn.TransformerEncoder 一致
torch.nn.TransformerDecoderLayer mindspore.nn.TransformerDecoderLayer 一致
torch.nn.TransformerEncoderLayer mindspore.nn.TransformerEncoderLayer 一致
torch.nn.TripletMarginLoss mindspore.nn.TripletMarginLoss 差异对比
torch.nn.Unflatten mindspore.nn.Unflatten 一致
torch.nn.Unfold mindspore.nn.Unfold 差异对比
torch.nn.Upsample mindspore.nn.Upsample 差异对比
torch.nn.ZeroPad2d mindspore.nn.ZeroPad2d 一致

torch.nn.functional

PyTorch 1.8.1 APIs MindSpore APIs 说明
torch.nn.functional.adaptive_avg_pool2d mindspore.ops.adaptive_avg_pool2d 一致
torch.nn.functional.adaptive_avg_pool3d mindspore.ops.adaptive_avg_pool3d 一致
torch.nn.functional.adaptive_max_pool1d mindspore.ops.adaptive_max_pool1d 差异对比
torch.nn.functional.adaptive_max_pool2d mindspore.ops.adaptive_max_pool2d 一致
torch.nn.functional.affine_grid mindspore.ops.affine_grid 一致
torch.nn.functional.avg_pool1d mindspore.ops.avg_pool1d 功能一致,参数名不同
torch.nn.functional.avg_pool2d mindspore.ops.avg_pool2d 功能一致,参数名不同
torch.nn.functional.avg_pool3d mindspore.ops.avg_pool3d 功能一致,参数名不同
torch.nn.functional.batch_norm mindspore.ops.batch_norm 一致
torch.nn.functional.binary_cross_entropy mindspore.ops.binary_cross_entropy 差异对比
torch.nn.functional.binary_cross_entropy_with_logits mindspore.ops.binary_cross_entropy_with_logits 功能一致,默认值不同
torch.nn.functional.conv1d mindspore.ops.conv1d 一致
torch.nn.functional.conv2d mindspore.ops.conv2d 一致
torch.nn.functional.conv3d mindspore.ops.conv3d 一致
torch.nn.functional.cosine_embedding_loss mindspore.ops.cosine_embedding_loss 一致
torch.nn.functional.cosine_similarity mindspore.ops.cosine_similarity 一致
torch.nn.functional.cross_entropy mindspore.ops.cross_entropy 一致
torch.nn.functional.ctc_loss mindspore.ops.ctc_loss 一致
torch.nn.functional.dropout mindspore.ops.dropout 一致
torch.nn.functional.dropout2d mindspore.ops.dropout2d 功能一致,数据类型不同
torch.nn.functional.dropout3d mindspore.ops.dropout3d 功能一致,数据类型不同
torch.nn.functional.elu mindspore.ops.elu 差异对比
torch.nn.functional.fold mindspore.ops.fold 差异对比
torch.nn.functional.gelu mindspore.ops.gelu 差异对比
torch.nn.functional.glu mindspore.ops.glu 功能一致,参数名不同
torch.nn.functional.grid_sample mindspore.ops.grid_sample 差异对比
torch.nn.functional.gumbel_softmax mindspore.ops.gumbel_softmax 一致
torch.nn.functional.hardshrink mindspore.ops.hardshrink 功能一致,参数名不同
torch.nn.functional.hardsigmoid mindspore.ops.hardsigmoid 一致
torch.nn.functional.hardswish mindspore.ops.hardswish 功能一致,参数名不同
torch.nn.functional.hardtanh mindspore.ops.hardtanh 一致
torch.nn.functional.hinge_embedding_loss mindspore.ops.hinge_embedding_loss 一致
torch.nn.functional.interpolate mindspore.ops.interpolate 差异对比
torch.nn.functional.kl_div mindspore.ops.kl_div 差异对比
torch.nn.functional.l1_loss mindspore.ops.l1_loss 一致
torch.nn.functional.leaky_relu mindspore.ops.leaky_relu 差异对比
torch.nn.functional.logsigmoid mindspore.ops.logsigmoid 功能一致,参数名不同
torch.nn.functional.log_softmax mindspore.ops.log_softmax 差异对比
torch.nn.functional.lp_pool1d mindspore.ops.lp_pool1d 功能一致,参数名不同
torch.nn.functional.lp_pool2d mindspore.ops.lp_pool2d 功能一致,参数名不同
torch.nn.functional.margin_ranking_loss mindspore.ops.margin_ranking_loss 一致
torch.nn.functional.max_pool3d mindspore.ops.max_pool3d 一致
torch.nn.functional.max_unpool1d mindspore.ops.max_unpool1d 一致
torch.nn.functional.max_unpool2d mindspore.ops.max_unpool2d 一致
torch.nn.functional.max_unpool3d mindspore.ops.max_unpool3d 一致
torch.nn.functional.mse_loss mindspore.ops.mse_loss 一致
torch.nn.functional.multi_margin_loss mindspore.ops.multi_margin_loss 一致
torch.nn.functional.multilabel_margin_loss mindspore.ops.multilabel_margin_loss 一致
torch.nn.functional.multilabel_soft_margin_loss mindspore.ops.multilabel_soft_margin_loss 一致
torch.nn.functional.nll_loss mindspore.ops.nll_loss 一致
torch.nn.functional.pad mindspore.ops.pad 差异对比
torch.nn.functional.pdist mindspore.ops.pdist 一致
torch.nn.functional.pixel_shuffle mindspore.ops.pixel_shuffle 一致
torch.nn.functional.pixel_unshuffle mindspore.ops.pixel_unshuffle 一致
torch.nn.functional.prelu mindspore.ops.prelu 功能一致,参数名不同
torch.nn.functional.relu mindspore.ops.relu 功能一致,参数名不同
torch.nn.functional.relu6 mindspore.ops.relu6 功能一致,参数名不同
torch.nn.functional.rrelu mindspore.ops.rrelu 功能一致,参数名不同
torch.nn.functional.selu mindspore.ops.selu 功能一致,参数名不同
torch.nn.functional.sigmoid mindspore.ops.sigmoid 功能一致,参数名不同
torch.nn.functional.silu mindspore.ops.silu 功能一致,参数名不同
torch.nn.functional.softmax mindspore.ops.softmax 差异对比
torch.nn.functional.softmin mindspore.ops.softmin 差异对比
torch.nn.functional.softsign mindspore.ops.softsign 功能一致,参数名不同
torch.nn.functional.smooth_l1_loss mindspore.ops.smooth_l1_loss 一致
torch.nn.functional.soft_margin_loss mindspore.nn.SoftMarginLoss 差异对比
torch.nn.functional.softshrink mindspore.ops.softshrink 功能一致,参数名不同
torch.nn.functional.tanh mindspore.ops.tanh 功能一致,参数名不同
torch.nn.functional.tanhshrink mindspore.ops.tanhshrink 一致
torch.nn.functional.threshold mindspore.ops.threshold 功能一致,参数名不同
torch.nn.functional.triplet_margin_loss mindspore.ops.triplet_margin_loss 差异对比
torch.nn.functional.upsample mindspore.ops.upsample 差异对比

torch.nn.Module

PyTorch 1.8.1 APIs MindSpore APIs 说明
torch.nn.Module.apply mindspore.nn.Cell.apply 一致

torch.nn.utils

PyTorch 1.8.1 APIs MindSpore APIs 说明
torch.nn.utils.clip_grad_value_ mindspore.ops.clip_by_value 差异对比

torch.Tensor

PyTorch 1.8.1 APIs MindSpore APIs 说明
torch.Tensor.abs mindspore.Tensor.abs 一致
torch.Tensor.absolute mindspore.Tensor.absolute 一致
torch.Tensor.acos mindspore.Tensor.acos 一致
torch.Tensor.acosh mindspore.Tensor.acosh 一致
torch.Tensor.add mindspore.Tensor.add 一致
torch.Tensor.addbmm mindspore.Tensor.addbmm 一致
torch.Tensor.addcdiv mindspore.Tensor.addcdiv 一致
torch.Tensor.addcmul mindspore.Tensor.addcmul 一致
torch.Tensor.addmm mindspore.Tensor.addmm 一致
torch.Tensor.addmv mindspore.Tensor.addmv 功能一致,参数名不同
torch.Tensor.addr mindspore.Tensor.addr 一致
torch.Tensor.all mindspore.Tensor.all 差异对比
torch.Tensor.amax mindspore.Tensor.amax 差异对比
torch.Tensor.amin mindspore.Tensor.amin 差异对比
torch.Tensor.any mindspore.Tensor.any 差异对比
torch.Tensor.arccos mindspore.Tensor.arccos 一致
torch.Tensor.arccosh mindspore.Tensor.arccosh 一致
torch.Tensor.argmax mindspore.Tensor.argmax 功能一致,参数名不同
torch.Tensor.angle mindspore.Tensor.angle 一致
torch.Tensor.arcsin mindspore.Tensor.arcsin 一致
torch.Tensor.arcsinh mindspore.Tensor.arcsinh 一致
torch.Tensor.arctan mindspore.Tensor.arctan 一致
torch.Tensor.arctanh mindspore.Tensor.arctanh 一致
torch.Tensor.argmin mindspore.Tensor.argmin 差异对比
torch.Tensor.argsort mindspore.Tensor.argsort 一致
torch.Tensor.asin mindspore.Tensor.asin 一致
torch.Tensor.asinh mindspore.Tensor.asinh 一致
torch.Tensor.atan mindspore.Tensor.atan 一致
torch.Tensor.atan2 mindspore.Tensor.atan2 一致
torch.Tensor.atanh mindspore.Tensor.atanh 一致
torch.Tensor.baddbmm mindspore.Tensor.baddbmm 一致
torch.Tensor.bernoulli mindspore.Tensor.bernoulli 差异对比
torch.Tensor.bincount mindspore.Tensor.bincount 一致
torch.Tensor.bitwise_and mindspore.Tensor.bitwise_and 一致
torch.Tensor.bitwise_or mindspore.Tensor.bitwise_or 一致
torch.Tensor.bitwise_xor mindspore.Tensor.bitwise_xor 一致
torch.Tensor.bmm mindspore.Tensor.bmm 功能一致,参数名不同
torch.Tensor.bool mindspore.Tensor.bool 功能一致,参数名不同
torch.Tensor.broadcast_to mindspore.Tensor.broadcast_to 差异对比
torch.Tensor.ceil mindspore.Tensor.ceil 一致
torch.Tensor.chunk mindspore.Tensor.chunk 一致
torch.Tensor.cholesky mindspore.Tensor.cholesky 一致
torch.Tensor.clamp mindspore.Tensor.clamp 一致
torch.Tensor.clip mindspore.Tensor.clip 差异对比
torch.Tensor.conj mindspore.Tensor.conj 一致
torch.Tensor.copysign mindspore.Tensor.copysign 一致
torch.Tensor.cos mindspore.Tensor.cos 一致
torch.Tensor.cosh mindspore.Tensor.cosh 一致
torch.Tensor.cross mindspore.Tensor.cross 一致
torch.Tensor.cummax mindspore.Tensor.cummax 功能一致,参数名不同
torch.Tensor.cummin mindspore.Tensor.cummin 功能一致,参数名不同
torch.Tensor.cumprod mindspore.Tensor.cumprod 一致
torch.Tensor.cumsum mindspore.Tensor.cumsum 一致
torch.Tensor.deg2rad mindspore.Tensor.deg2rad 一致
torch.Tensor.diag mindspore.Tensor.diag) 差异对比
torch.Tensor.diagflat mindspore.Tensor.diagflat 一致
torch.Tensor.diff mindspore.Tensor.diff 功能一致,参数名不同
torch.Tensor.digamma mindspore.Tensor.digamma 一致
torch.Tensor.dim mindspore.Tensor.ndimension 一致
torch.Tensor.div mindspore.Tensor.div 功能一致,参数名不同
torch.Tensor.divide mindspore.Tensor.divide 一致
torch.Tensor.equal mindspore.Tensor.equal 一致
torch.Tensor.erf mindspore.Tensor.erf 一致
torch.Tensor.erfc mindspore.Tensor.erfc 一致
torch.Tensor.erfinv mindspore.Tensor.erfinv 一致
torch.Tensor.exp mindspore.Tensor.exp 一致
torch.Tensor.expand_as mindspore.Tensor.expand_as 功能一致,参数名不同
torch.Tensor.expm1 mindspore.Tensor.expm1 一致
torch.Tensor.flip mindspore.Tensor.flip 一致
torch.Tensor.fliplr mindspore.Tensor.fliplr 一致
torch.Tensor.flipud mindspore.Tensor.flipud 一致
torch.Tensor.float mindspore.Tensor.float 功能一致,参数名不同
torch.Tensor.float_power mindspore.Tensor.float_power 差异对比
torch.Tensor.floor mindspore.Tensor.floor 一致
torch.Tensor.fmod mindspore.Tensor.fmod 一致
torch.Tensor.frac mindspore.Tensor.frac 功能一致,参数名不同
torch.Tensor.gather mindspore.Tensor.gather_elements 一致
torch.Tensor.ge mindspore.Tensor.ge 功能一致,参数名不同
torch.Tensor.geqrf mindspore.Tensor.geqrf 一致
torch.Tensor.ger mindspore.Tensor.ger 差异对比
torch.Tensor.greater mindspore.Tensor.greater 功能一致,参数名不同
torch.Tensor.greater_equal mindspore.Tensor.greater_equal 功能一致,参数名不同
torch.Tensor.gt mindspore.Tensor.gt 功能一致,参数名不同
torch.Tensor.H mindspore.Tensor.H 一致
torch.Tensor.half mindspore.Tensor.half 功能一致,参数名不同
torch.Tensor.hardshrink mindspore.Tensor.hardshrink 一致
torch.Tensor.heaviside mindspore.Tensor.heaviside 一致
torch.Tensor.histc mindspore.Tensor.histc 一致
torch.Tensor.hypot mindspore.Tensor.hypot 一致
torch.Tensor.i0 mindspore.Tensor.i0 一致
torch.Tensor.igamma mindspore.Tensor.igamma 一致
torch.Tensor.igammac mindspore.Tensor.igammac 一致
torch.Tensor.imag mindspore.Tensor.imag 一致
torch.Tensor.index_add mindspore.Tensor.index_add 功能一致,参数名不同
torch.Tensor.index_fill mindspore.Tensor.index_fill 一致
torch.Tensor.index_select mindspore.Tensor.index_select 功能一致,参数名不同
torch.Tensor.inner mindspore.Tensor.inner 功能一致,参数名不同
torch.Tensor.int mindspore.Tensor.int 功能一致,参数名不同
torch.Tensor.inverse mindspore.Tensor.inverse 一致
torch.Tensor.isclose mindspore.Tensor.isclose 功能一致,参数名不同
torch.Tensor.isfinite mindspore.Tensor.isfinite 一致
torch.Tensor.isinf mindspore.Tensor.isinf 一致
torch.Tensor.isnan mindspore.Tensor.isnan 一致
torch.Tensor.is_complex mindspore.Tensor.is_complex 一致
torch.Tensor.is_floating_point mindspore.Tensor.is_floating_point 一致
torch.Tensor.is_signed mindspore.Tensor.is_signed 一致
torch.Tensor.isneginf mindspore.Tensor.isneginf 一致
torch.Tensor.isposinf mindspore.Tensor.isposinf 一致
torch.Tensor.isreal mindspore.Tensor.isreal 一致
torch.Tensor.item mindspore.Tensor.item 差异对比
torch.Tensor.lcm mindspore.Tensor.lcm 一致
torch.Tensor.ldexp mindspore.Tensor.ldexp 一致
torch.Tensor.le mindspore.Tensor.le 功能一致,参数名不同
torch.Tensor.lerp mindspore.Tensor.lerp 一致
torch.Tensor.less mindspore.Tensor.less 功能一致,参数名不同
torch.Tensor.less_equal mindspore.Tensor.less_equal 功能一致,参数名不同
torch.Tensor.log mindspore.Tensor.log 一致
torch.Tensor.log10 mindspore.Tensor.log10 一致
torch.Tensor.log1p mindspore.Tensor.log1p 一致
torch.Tensor.log2 mindspore.Tensor.log2 一致
torch.Tensor.logaddexp mindspore.Tensor.logaddexp 一致
torch.Tensor.logaddexp2 mindspore.Tensor.logaddexp2 一致
torch.Tensor.logdet mindspore.Tensor.logdet 一致
torch.Tensor.logical_and mindspore.Tensor.logical_and 一致
torch.Tensor.logical_not mindspore.Tensor.logical_not 一致
torch.Tensor.logical_or mindspore.Tensor.logical_or 一致
torch.Tensor.logical_xor mindspore.Tensor.logical_xor 一致
torch.Tensor.logit mindspore.Tensor.logit 功能一致,参数名不同
torch.Tensor.logsumexp mindspore.Tensor.logsumexp 一致
torch.Tensor.long mindspore.Tensor.long 功能一致,参数名不同
torch.Tensor.lt mindspore.Tensor.lt 一致
torch.Tensor.masked_fill mindspore.Tensor.masked_fill 一致
torch.Tensor.masked_select mindspore.Tensor.masked_select 一致
torch.Tensor.matmul mindspore.Tensor.matmul 一致
torch.Tensor.max mindspore.Tensor.max 差异对比
torch.Tensor.maximum mindspore.Tensor.maximum 功能一致,参数名不同
torch.Tensor.mean mindspore.Tensor.mean 功能一致,参数名不同
torch.Tensor.median mindspore.Tensor.median 差异对比
torch.Tensor.mH mindspore.Tensor.mH 一致
torch.Tensor.min mindspore.Tensor.min 差异对比
torch.Tensor.minimum mindspore.Tensor.minimum 功能一致,参数名不同
torch.Tensor.mm mindspore.Tensor.mm 一致
torch.Tensor.moveaxis mindspore.Tensor.moveaxis 功能一致
torch.Tensor.movedim mindspore.Tensor.movedim 一致
torch.Tensor.msort mindspore.Tensor.msort 一致
torch.Tensor.mT mindspore.Tensor.mT 一致
torch.Tensor.mul mindspore.Tensor.mul 一致
torch.Tensor.multinomial mindspore.Tensor.multinomial 差异对比
torch.Tensor.multiply mindspore.Tensor.multiply 一致
torch.Tensor.mvlgamma mindspore.Tensor.mvlgamma 一致
torch.Tensor.nan_to_num mindspore.Tensor.nan_to_num 功能一致,数据类型不同
torch.Tensor.nansum mindspore.Tensor.nansum 一致
torch.Tensor.narrow mindspore.Tensor.narrow 一致
torch.Tensor.ndim mindspore.Tensor.ndim 一致
torch.Tensor.ndimension mindspore.Tensor.ndimension 一致
torch.Tensor.ne mindspore.Tensor.ne 一致,参数名不同
torch.Tensor.neg mindspore.Tensor.neg 一致
torch.Tensor.negative mindspore.Tensor.negative 一致
torch.Tensor.nelement mindspore.Tensor.nelement 一致
torch.Tensor.new_ones mindspore.Tensor.new_ones 一致
torch.Tensor.new_zeros mindspore.Tensor.new_zeros 一致
torch.Tensor.nextafter mindspore.Tensor.nextafter 一致
torch.Tensor.norm mindspore.Tensor.norm torch.Tensor.norm即将弃用,mindspore.Tensor.norm对标torch.linalg.norm,功能完全一致
torch.Tensor.nonzero mindspore.Tensor.nonzero 一致
torch.Tensor.not_equal mindspore.Tensor.not_equal 一致
torch.Tensor.numel mindspore.Tensor.numel 一致
torch.Tensor.numpy mindspore.Tensor.asnumpy 一致
torch.Tensor.permute mindspore.Tensor.permute 一致
torch.Tensor.pow mindspore.Tensor.pow 功能一致,参数名不同
torch.Tensor.prod mindspore.Tensor.prod 差异对比
torch.Tensor.qr mindspore.Tensor.qr 一致
torch.Tensor.ravel mindspore.Tensor.ravel 一致
torch.Tensor.real mindspore.Tensor.real 一致
torch.Tensor.reciprocal mindspore.Tensor.reciprocal 一致
torch.Tensor.remainder mindspore.Tensor.remainder 一致
torch.Tensor.renorm mindspore.Tensor.renorm 差异对比
torch.Tensor.rad2deg mindspore.Tensor.rad2deg 一致
torch.Tensor.repeat mindspore.Tensor.tile 功能一致,参数名不同
torch.Tensor.repeat_interleave mindspore.Tensor.repeat_interleave 一致
torch.Tensor.reshape mindspore.Tensor.reshape 一致
torch.Tensor.reshape_as mindspore.Tensor.reshape_as 一致
torch.Tensor.round mindspore.Tensor.round 一致
torch.Tensor.rot90 mindspore.Tensor.rot90 一致
torch.Tensor.rsqrt mindspore.Tensor.rsqrt 一致
torch.Tensor.scatter_ mindspore.ops.tensor_scatter_elements 差异对比
torch.Tensor.sum_to_size mindspore.Tensor.sum_to_size 一致
torch.Tensor.scatter mindspore.Tensor.scatter 差异对比
torch.Tensor.sgn mindspore.Tensor.sgn 一致
torch.Tensor.short mindspore.Tensor.short 一致
torch.Tensor.sigmoid mindspore.Tensor.sigmoid 一致
torch.Tensor.sign mindspore.Tensor.sign 一致
torch.Tensor.signbit mindspore.Tensor.signbit 一致
torch.Tensor.sin mindspore.Tensor.sin 一致
torch.Tensor.sinc mindspore.Tensor.sinc 一致
torch.Tensor.sinh mindspore.Tensor.sinh 一致
torch.Tensor.size mindspore.Tensor.shape 一致
torch.Tensor.slogdet mindspore.Tensor.slogdet 一致
torch.Tensor.sort mindspore.Tensor.sort 一致
torch.Tensor.split mindspore.Tensor.split 功能一致,参数名不同
torch.Tensor.sqrt mindspore.Tensor.sqrt 一致
torch.Tensor.square mindspore.Tensor.square 一致
torch.Tensor.squeeze mindspore.Tensor.squeeze 一致
torch.Tensor.std mindspore.Tensor.std 差异对比
torch.Tensor.sub mindspore.Tensor.subtract 一致
torch.Tensor.subtract mindspore.Tensor.subtract 一致
torch.Tensor.svd mindspore.Tensor.svd 差异对比
torch.Tensor.swapaxes mindspore.Tensor.swapaxes 一致
torch.Tensor.swapdims mindspore.Tensor.swapdims 一致
torch.Tensor.T mindspore.Tensor.T 一致
torch.Tensor.t mindspore.Tensor.t 一致
torch.Tensor.take mindspore.Tensor.take 一致
torch.Tensor.tan mindspore.Tensor.tan 一致
torch.Tensor.tanh mindspore.Tensor.tanh 一致
torch.Tensor.trace mindspore.Tensor.trace 一致
torch.Tensor.transpose mindspore.Tensor.swapaxes 一致
torch.Tensor.tile mindspore.Tensor.tile 功能一致,参数名不同
torch.Tensor.to mindspore.Tensor.to 功能一致,参数名不同
torch.Tensor.topk mindspore.Tensor.topk 一致
torch.Tensor.tril mindspore.Tensor.tril 功能一致,参数名不同
torch.Tensor.tensor_split mindspore.Tensor.tensor_split 一致
torch.Tensor.true_divide mindspore.Tensor.true_divide 一致
torch.Tensor.trunc mindspore.Tensor.trunc 一致
torch.Tensor.unbind mindspore.Tensor.unbind 一致
torch.Tensor.unique_consecutive mindspore.Tensor.unique_consecutive 一致
torch.Tensor.unsqueeze mindspore.Tensor.unsqueeze 一致
torch.Tensor.var mindspore.Tensor.var 差异对比
torch.Tensor.view mindspore.Tensor.view 一致
torch.Tensor.view_as mindspore.Tensor.view_as 一致
torch.Tensor.where mindspore.Tensor.where 一致
torch.Tensor.xlogy mindspore.Tensor.xlogy 功能一致,参数名不同

torchvision

TorchVision 0.10.0 APIs MindSpore APIs 说明
torchvision.datasets.CelebA mindspore.dataset.CelebADataset
torchvision.datasets.CIFAR10 mindspore.dataset.Cifar10Dataset
torchvision.datasets.CIFAR100 mindspore.dataset.Cifar100Dataset
torchvision.datasets.CocoDetection mindspore.dataset.CocoDataset 差异对比
torchvision.datasets.ImageFolder mindspore.dataset.ImageFolderDataset
torchvision.datasets.MNIST mindspore.dataset.MnistDataset
torchvision.datasets.VOCDetection mindspore.dataset.VOCDataset 差异对比
torchvision.datasets.VOCSegmentation mindspore.dataset.VOCDataset 差异对比
torchvision.ops.nms mindspore.ops.NMSWithMask
torchvision.ops.roi_align mindspore.ops.ROIAlign
torchvision.transforms.CenterCrop mindspore.dataset.vision.CenterCrop
torchvision.transforms.ColorJitter mindspore.dataset.vision.RandomColorAdjust
torchvision.transforms.Compose mindspore.dataset.transforms.Compose
torchvision.transforms.ConvertImageDtype mindspore.dataset.transforms.TypeCast 差异对比
torchvision.transforms.FiveCrop mindspore.dataset.vision.FiveCrop
torchvision.transforms.GaussianBlur mindspore.dataset.vision.GaussianBlur
torchvision.transforms.Grayscale mindspore.dataset.vision.Grayscale
torchvision.transforms.LinearTransformation mindspore.dataset.vision.LinearTransformation
torchvision.transforms.Normalize mindspore.dataset.vision.Normalize
torchvision.transforms.Pad mindspore.dataset.vision.Pad
torchvision.transforms.RandomAffine mindspore.dataset.vision.RandomAffine
torchvision.transforms.RandomApply mindspore.dataset.transforms.RandomApply
torchvision.transforms.RandomChoice mindspore.dataset.transforms.RandomChoice
torchvision.transforms.RandomCrop mindspore.dataset.vision.RandomCrop
torchvision.transforms.RandomErasing mindspore.dataset.vision.RandomErasing
torchvision.transforms.RandomGrayscale mindspore.dataset.vision.RandomGrayscale
torchvision.transforms.RandomHorizontalFlip mindspore.dataset.vision.RandomHorizontalFlip
torchvision.transforms.RandomOrder mindspore.dataset.transforms.RandomOrder
torchvision.transforms.RandomPerspective mindspore.dataset.vision.RandomPerspective
torchvision.transforms.RandomPosterize mindspore.dataset.vision.RandomPosterize
torchvision.transforms.RandomResizedCrop mindspore.dataset.vision.RandomResizedCrop
torchvision.transforms.RandomRotation mindspore.dataset.vision.RandomRotation
torchvision.transforms.RandomSolarize mindspore.dataset.vision.RandomSolarize 差异对比
torchvision.transforms.RandomVerticalFlip mindspore.dataset.vision.RandomVerticalFlip
torchvision.transforms.Resize mindspore.dataset.vision.Resize
torchvision.transforms.TenCrop mindspore.dataset.vision.TenCrop
torchvision.transforms.ToPILImage mindspore.dataset.vision.ToPIL 差异对比
torchvision.transforms.ToTensor mindspore.dataset.vision.ToTensor 差异对比
TorchVision 0.14.0 APIs MindSpore APIs 说明
torchvision.ops.deform_conv2d mindspore.ops.deformable_conv2d 差异对比