简体中文 | English
MSAdapter接口 | 状态 | 备注 |
---|---|---|
torch.is_tensor | 支持 | |
torch.is_floating_point | 支持 | |
torch.arange | 支持 | |
torch.cat | 支持 | |
torch.tensor | 支持 | |
torch.as_tensor | 支持 | |
torch.from_numpy | 支持 | |
torch.frombuffer | 支持 | require_grad暂不支持 |
torch.permute | 支持 | |
torch.bitwise_left_shift | 支持 | |
torch.bitwise_right_shift | 支持 | |
torch.nan_to_num | 支持 | |
torch.range | 支持 | 不支持Ascend |
torch.linspace | 支持 | |
torch.logspace | 支持 | |
torch.eye | 支持 | |
torch.empty | 支持 | |
torch.empty_like | 支持 | |
torch.full | 支持 | |
torch.full_like | 支持 | |
torch.polar | 支持 | |
torch.concat | 支持 | |
torch.column_stack | 支持 | |
torch.gather | 支持 | |
torch.is_complex | 支持 | |
torch.hstack | 支持 | |
torch.index_select | 支持 | |
torch.masked_select | 支持 | |
torch.movedim | 支持 | |
torch.moveaxis | 支持 | |
torch.narrow | 支持 | |
torch.nonzero | 支持 | |
torch.numel | 支持 | |
torch.reshape | 支持 | |
torch.row_stack | 支持 | |
torch.select | 支持 | |
torch.zeros | 支持 | |
torch.squeeze | 支持 | |
torch.stack | 支持 | |
torch.swapaxes | 支持 | |
torch.swapdims | 支持 | |
torch.zeros_like | 支持 | |
torch.take | 支持 | |
torch.ones | 支持 | |
torch.tile | 支持 | |
torch.transpose | 支持 | |
torch.unbind | 支持 | |
torch.unsqueeze | 支持 | |
torch.ones_like | 支持 | |
torch.vstack | 支持 | |
torch.heaviside | 支持 | |
torch.seed | 支持 | |
torch.initial_seed | 支持 | |
torch.rand | 支持 | |
torch.randn | 支持 | |
torch.abs | 支持 | |
torch.absolute | 支持 | |
torch.acos | 支持 | |
torch.adjoint | 支持 | |
torch.acosh | 支持 | |
torch.arccosh | 支持 | |
torch.add | 支持 | |
torch.addcdiv | 支持 | |
torch.addcmul | 支持 | |
torch.dsplit | 支持 | |
torch.asin | 支持 | |
torch.arcsin | 支持 | |
torch.asinh | 支持 | |
torch.arcsinh | 支持 | |
torch.atan | 支持 | |
torch.arctan | 支持 | |
torch.atanh | 支持 | |
torch.arctanh | 支持 | |
torch.atan2 | 支持 | |
torch.arctan2 | 支持 | |
torch.bitwise_not | 支持 | |
torch.bitwise_and | 支持 | |
torch.bitwise_or | 支持 | |
torch.bitwise_xor | 支持 | |
torch.hsplit | 支持 | |
torch.split | 支持 | |
torch.ceil | 支持 | |
torch.t | 支持 | |
torch.tensor_split | 支持 | |
torch.conj_physical | 支持 | |
torch.copysign | 支持 | |
torch.cos | 支持 | |
torch.cosh | 支持 | |
torch.deg2rad | 支持 | |
torch.device | 支持 | |
torch.div | 支持 | |
torch.divide | 支持 | |
torch.erf | 支持 | |
torch.erfc | 支持 | |
torch.erfinv | 支持 | |
torch.exp | 支持 | |
torch.exp2 | 支持 | |
torch.expm1 | 支持 | |
torch.fix | 支持 | |
torch.vsplit | 支持 | |
torch.floor | 支持 | |
torch.floor_divide | 支持 | |
torch.where | 支持 | |
torch.frac | 支持 | |
torch.frexp | 支持 | |
torch.finfo | 支持 | |
torch.iinfo | 支持 | |
torch.index_select | 支持 | |
torch.ldexp | 支持 | |
torch.lerp | 支持 | |
torch.arccos | 支持 | |
torch.log | 支持 | |
torch.angle | 支持 | |
torch.log1p | 支持 | |
torch.clamp | 支持 | |
torch.logaddexp | 支持 | |
torch.logaddexp2 | 支持 | |
torch.logical_not | 支持 | |
torch.logical_or | 支持 | |
torch.logit | 支持 | |
torch.clip | 支持 | |
torch.float_power | 支持 | |
torch.igammac | 支持 | |
torch.mul | 支持 | |
torch.fmod | 支持 | |
torch.lgamma | 支持 | |
torch.neg | 支持 | |
torch.log10 | 支持 | |
torch.nextafter | 支持 | |
torch.positive | 支持 | |
torch.pow | 支持 | |
torch.rad2deg | 支持 | |
torch.log2 | 支持 | |
torch.hypot | 支持 | |
torch.remainder | 支持 | |
torch.round | 支持 | |
torch.sigmoid | 支持 | |
torch.multiply | 支持 | |
torch.negative | 支持 | |
torch.sin | 支持 | |
torch.reciprocal | 支持 | |
torch.sinh | 支持 | |
torch.sqrt | 支持 | |
torch.square | 支持 | |
torch.sub | 支持 | |
torch.rsqrt | 支持 | |
torch.tan | 支持 | |
torch.tanh | 支持 | |
torch.sign | 支持 | |
torch.trunc | 支持 | |
torch.xlogy | 支持 | |
torch.amax | 支持 | |
torch.amin | 支持 | |
torch.aminmax | 支持 | |
torch.all | 支持 | |
torch.any | 支持 | |
torch.min | 支持 | |
torch.dist | 支持 | |
torch.logsumexp | 支持 | |
torch.nanmean | 支持 | |
torch.nansum | 支持 | |
torch.prod | 支持 | |
torch.std | 支持 | |
torch.sgn | 支持 | |
torch.unique_consecutive | 支持 | |
torch.var | 支持 | |
torch.count_nonzero | 支持 | |
torch.allclose | 支持 | |
torch.signbit | 支持 | |
torch.eq | 支持 | |
torch.equal | 支持 | |
torch.ge | 支持 | |
torch.greater_equal | 支持 | |
torch.gt | 支持 | |
torch.greater | 支持 | |
torch.isclose | 支持 | |
torch.isfinite | 支持 | |
torch.isinf | 支持 | |
torch.isposinf | 支持 | |
torch.isneginf | 支持 | |
torch.isnan | 支持 | |
torch.isreal | 支持 | |
torch.le | 支持 | |
torch.less_equal | 支持 | |
torch.lt | 支持 | |
torch.less | 支持 | |
torch.maximum | 支持 | |
torch.minimum | 支持 | |
torch.ne | 支持 | |
torch.sinc | 支持 | |
torch.subtract | 支持 | |
torch.topk | 支持 | |
torch.true_divide | 支持 | |
torch.atleast_1d | 支持 | |
torch.atleast_2d | 支持 | |
torch.atleast_3d | 支持 | |
torch.block_diag | 支持 | |
torch.broadcast_to | 支持 | |
torch.cdist | 支持 | |
torch.corrcoef | 支持 | |
torch.cov | 支持 | |
torch.cummin | 支持 | |
torch.cumprod | 支持 | |
torch.cumsum | 支持 | |
torch.cumsum | 支持 | |
torch.diag | 支持 | |
torch.diagflat | 支持 | |
torch.diagonal | 支持 | |
torch.diff | 支持 | |
torch.flatten | 支持 | |
torch.flip | 支持 | |
torch.flipud | 支持 | |
torch.histc | 支持 | 仅支持Ascend GPU |
torch.meshgrid | 支持 | |
torch.ravel | 支持 | |
torch.not_equal | 支持 | |
torch.trace | 支持 | |
torch.tril | 支持 | |
torch.triu | 支持 | |
torch.sort | 支持 | |
torch.msort | 支持 | |
torch.addmv | 支持 | |
torch.addr | 支持 | |
torch.bincount | 支持 | |
torch.bmm | 支持 | |
torch.cholesky | 支持 | |
torch.cholesky_inverse | 支持 | |
torch.dot | 支持 | |
torch.repeat_interleave | 支持 | |
torch.addbmm | 支持 | |
torch.det | 支持 | |
torch.addmm | 支持 | |
torch.matmul | 支持 | |
torch.mv | 支持 | |
torch.outer | 支持 | |
torch.vdot | 支持 | |
torch._assert | 支持 | |
torch.inner | 支持 | |
torch.logdet | 支持 | |
torch.mm | 支持 | |
torch.cuda.is_available | 支持 | |
torch.ByteTensor | 支持 | |
torch.CharTensor | 支持 | |
torch.IntTensor | 支持 | |
torch.HalfTensor | 支持 | |
torch.FloatTensor | 支持 | |
torch.DoubleTensor | 支持 | |
torch.ByteStorage | 支持 | |
torch.as_strided | 支持 | |
torch.view_as_real | 支持 | |
torch.scatter | 不支持 | |
torch.manual_seed | 支持 | |
torch.bernoulli | 支持 | |
torch.multinomial | 支持 | 暂不支持传入Generator,输入只支持float32, Ascend上暂不支持 |
torch.randint | 支持 | 暂不支持传入Generator |
torch.randperm | 支持 | 暂不支持传入Generator |
torch.digamma | 支持 | Ascend暂时不支持 |
torch.fft | 支持 | |
torch.finfo | 支持 | |
torch.gradient | 支持 | |
torch.imag | 支持 | 不支持视图, Ascend上暂不支持 |
torch.logical_and | 支持 | Ascend上暂不支持 |
torch.logical_xor | 支持 | Ascend上暂不支持 |
torch.igamma | 支持 | Ascend上暂不支持 |
torch.mvlgamma | 支持 | Ascend上暂不支持 |
torch.i0 | 支持 | Ascend上暂不支持 |
torch.real | 支持 | 不支持视图, Ascend上暂不支持 |
torch.argmax | 支持 | 暂无keepdims参数 |
torch.argmin | 支持 | 暂无keepdims参数 |
torch.max | 支持 | 不支持other,不支持graph模式 |
torch.mean | 支持 | |
torch.median | 支持 | Ascend上暂不支持 |
torch.norm | 支持 | |
torch.unique | 支持 | |
torch.bartlett_window | 支持 | Ascend上暂不支持 |
torch.sum | 支持 | 不支持graph模式 |
torch.hann_window | 支持 | |
torch.argsort | 支持 | |
torch.cross | 支持 | Ascend上暂不支持 |
torch.cummax | 支持 | Ascend上暂不支持 |
torch.einsum | 支持 | 仅支持GPU |
torch.fliplr | 支持 | |
torch.hamming_window | 支持 | |
torch.svd | 支持 | Ascend上暂不支持 |
torch.fmax | 支持 | GPU上暂不支持 |
torch.fmin | 支持 | GPU上暂不支持 |
torch.histc | 支持 | |
torch.inverse | 支持 | 不支持Ascend |
torch.poisson | 支持 | |
torch.matrix_power | 支持 | 仅支持CPU |
MSAdapter接口 | 状态 | 备注 |
---|---|---|
Tensor.mm | 支持 | |
Tensor.msort | 支持 | |
Tensor.abs | 支持 | |
Tensor.absolute | 支持 | |
Tensor.acos | 支持 | |
Tensor.acosh | 支持 | |
Tensor.new | 支持 | |
Tensor.new_tensor | 支持 | |
Tensor.new_full | 支持 | |
Tensor.new_empty | 支持 | |
Tensor.new_ones | 支持 | |
Tensor.new_zeros | 支持 | |
Tensor.is_cuda | 支持 | |
Tensor.ndim | 支持 | |
Tensor.add | 支持 | |
Tensor.addbmm | 支持 | |
Tensor.addcdiv | 支持 | |
Tensor.addcmul | 支持 | |
Tensor.addmm | 支持 | |
Tensor.addmv | 支持 | |
Tensor.addr | 支持 | |
Tensor.all | 支持 | |
Tensor.allclose | 支持 | |
Tensor.amax | 支持 | |
Tensor.amin | 支持 | |
Tensor.aminmax | 支持 | |
Tensor.any | 支持 | |
Tensor.arccos | 支持 | |
Tensor.arccosh | 支持 | |
Tensor.arcsin | 支持 | |
Tensor.arcsinh | 支持 | |
Tensor.arctan | 支持 | |
Tensor.arctan2 | 支持 | |
Tensor.arctanh | 支持 | |
Tensor.asin | 支持 | |
Tensor.asinh | 支持 | |
Tensor.atan | 支持 | |
Tensor.atan2 | 支持 | |
Tensor.atanh | 支持 | |
Tensor.baddbmm | 支持 | |
Tensor.bincount | 支持 | |
Tensor.bitwise_and | 支持 | |
Tensor.bitwise_left_shift | 支持 | |
Tensor.bitwise_not | 支持 | |
Tensor.bitwise_or | 支持 | |
Tensor.bitwise_right_shift | 支持 | |
Tensor.bitwise_xor | 支持 | |
Tensor.bmm | 支持 | |
Tensor.bool | 支持 | 不支持memory_format参数 |
Tensor.broadcast_to | 支持 | |
Tensor.byte | 支持 | |
Tensor.ceil | 支持 | |
Tensor.char | 支持 | |
Tensor.cholesky | 支持 | |
Tensor.cholesky_inverse | 支持 | |
Tensor.clamp | 支持 | |
Tensor.clip | 支持 | |
Tensor.clone | 支持 | |
Tensor.conj | 支持 | |
Tensor.copysign | 支持 | |
Tensor.cos | 支持 | |
Tensor.cosh | 支持 | |
Tensor.count_nonzero | 支持 | |
Tensor.cpu | 支持 | |
Tensor.cummax | 支持 | |
Tensor.cummin | 支持 | |
Tensor.cumprod | 支持 | |
Tensor.cumsum | 支持 | |
Tensor.data | 支持 | |
Tensor.deg2rad | 支持 | |
Tensor.detach | 支持 | |
Tensor.diag | 支持 | |
Tensor.diagflat | 支持 | |
Tensor.diagonal | 支持 | |
Tensor.diff | 支持 | |
Tensor.dim | 支持 | |
Tensor.dist | 支持 | |
Tensor.divide | 支持 | |
Tensor.dot | 支持 | |
Tensor.double | 支持 | |
Tensor.dsplit | 支持 | |
Tensor.eq | 支持 | |
Tensor.equal | 支持 | |
Tensor.erf | 支持 | |
Tensor.erfc | 支持 | |
Tensor.erfinv | 支持 | |
Tensor.exp | 支持 | |
Tensor.expand_as | 支持 | |
Tensor.expm1 | 支持 | |
Tensor.fix | 支持 | |
Tensor.flatten | 支持 | |
Tensor.flip | 支持 | |
Tensor.flipud | 支持 | |
Tensor.float_power | 支持 | |
Tensor.floor | 支持 | |
Tensor.fmod | 支持 | |
Tensor.gather | 支持 | |
Tensor.ge | 支持 | |
Tensor.ger | 支持 | |
Tensor.greater | 支持 | |
Tensor.greater_equal | 支持 | |
Tensor.gt | 支持 | |
Tensor.half | 支持 | |
Tensor.hardshrink | 支持 | |
Tensor.heaviside | 支持 | |
Tensor.hsplit | 支持 | |
Tensor.hypot | 支持 | |
Tensor.index_select | 支持 | |
Tensor.int | 支持 | |
Tensor.is_complex | 支持 | |
Tensor.isclose | 支持 | |
Tensor.isfinite | 支持 | |
Tensor.isinf | 支持 | |
Tensor.isnan | 支持 | |
Tensor.isneginf | 支持 | |
Tensor.isposinf | 支持 | |
Tensor.isreal | 支持 | |
Tensor.is_tensor | 支持 | |
Tensor.item | 支持 | |
Tensor.le | 支持 | |
Tensor.less | 支持 | |
Tensor.less_equal | 支持 | |
Tensor.log | 支持 | |
Tensor.log10 | 支持 | |
Tensor.log1p | 支持 | |
Tensor.log2 | 支持 | |
Tensor.logaddexp | 支持 | |
Tensor.logdet | 支持 | |
Tensor.logical_not | 支持 | |
Tensor.logical_or | 支持 | |
Tensor.logsumexp | 支持 | |
Tensor.long | 支持 | |
Tensor.lt | 支持 | |
Tensor.masked_fill | 支持 | |
Tensor.matmul | 支持 | |
Tensor.max | 支持 | |
Tensor.maximum | 支持 | |
Tensor.mean | 支持 | |
Tensor.min | 支持 | |
Tensor.fmax | 支持 | GPU暂时不支持 |
Tensor.fmin | 支持 | GPU暂时不支持 |
Tensor.histc | 支持 | |
Tensor.minimum | 支持 | |
Tensor.moveaxis | 支持 | |
Tensor.movedim | 支持 | |
Tensor.mul | 支持 | |
Tensor.multiply | 支持 | |
Tensor.nanmean | 支持 | |
Tensor.nansum | 支持 | |
Tensor.narrow | 支持 | |
Tensor.ndimension | 支持 | |
Tensor.ne | 支持 | |
Tensor.neg | 支持 | |
Tensor.negative | 支持 | |
Tensor.nonzero | 支持 | |
Tensor.not_equal | 支持 | |
Tensor.numel | 支持 | |
Tensor.numpy | 支持 | |
Tensor.permute | 支持 | |
Tensor.pow | 支持 | |
Tensor.prod | 支持 | |
Tensor.rad2deg | 支持 | |
Tensor.ravel | 支持 | |
Tensor.random_ | 支持 | |
Tensor.reciprocal | 支持 | |
Tensor.remainder | 支持 | |
Tensor.repeat | 支持 | |
Tensor.repeat_interleave | 支持 | |
Tensor.reshape | 支持 | |
Tensor.reshape_as | 支持 | |
Tensor.resize_as_ | 支持 | |
Tensor.round | 支持 | |
Tensor.rsqrt_ | 支持 | |
Tensor.rsqrt | 支持 | |
Tensor.select | 支持 | |
Tensor.short | 支持 | |
Tensor.sigmoid | 支持 | |
Tensor.sign | 支持 | |
Tensor.signbit | 支持 | |
Tensor.sin | 支持 | |
Tensor.sinc | 支持 | |
Tensor.sinh | 支持 | |
Tensor.size | 支持 | |
Tensor.sort | 支持 | |
Tensor.split | 支持 | |
Tensor.sqrt | 支持 | |
Tensor.square | 支持 | |
Tensor.squeeze | 支持 | |
Tensor.stride | 支持 | |
Tensor.sub | 支持 | |
Tensor.subtract | 支持 | |
Tensor.sum | 支持 | |
Tensor.swapaxes | 支持 | |
Tensor.swapdims | 支持 | |
Tensor.T | 支持 | |
Tensor.t | 支持 | |
Tensor.H | 支持 | |
Tensor.take | 支持 | |
Tensor.tan | 支持 | |
Tensor.tanh | 支持 | |
Tensor.tensor_split | 支持 | |
Tensor.tile | 支持 | |
Tensor.tolist | 支持 | |
Tensor.topk | 支持 | |
Tensor.trace | 支持 | |
Tensor.transpose | 支持 | |
Tensor.tril | 支持 | |
Tensor.tril_ | 支持 | |
Tensor.triu | 支持 | |
Tensor.triu_ | 支持 | |
Tensor.true_divide | 支持 | |
Tensor.true_divide_ | 支持 | |
Tensor.trunc | 支持 | |
Tensor.type | 支持 | |
Tensor.type_as | 支持 | |
Tensor.unbind | 支持 | |
Tensor.uniform_ | 支持 | |
Tensor.unique_consecutive | 支持 | |
Tensor.unsqueeze | 支持 | |
Tensor.var | 支持 | |
Tensor.vdot | 支持 | |
Tensor.view_as | 支持 | |
Tensor.vsplit | 支持 | |
Tensor.xlogy_ | 支持 | |
Tensor.argmin | 支持 | 暂不支持keepdim |
Tensor.argsort | 支持 | |
Tensor.as_strided | 支持 | 不支持视图 |
Tensor.bernoulli | 支持 | 只支持GPU |
Tensor.bernoulli_ | 支持 | 只支持GPU |
Tensor.chunk | 支持 | |
Tensor.contiguous | 支持 | |
Tensor.cross | 支持 | Ascend上暂不支持 |
Tensor.cuda | 支持 | |
Tensor.det | 支持 | Ascend上暂不支持 |
Tensor.digamma | 支持 | Ascend上暂不支持 |
Tensor.div | 支持 | |
Tensor.expand | 支持 | 类型限制,只支持Tensor[Float16], Tensor[Float32], Tensor[Int32], Tensor[Int8], Tensor[UInt8] |
Tensor.fliplr | 支持 | Ascend上暂不支持 |
Tensor.float | 支持 | 不支持memory_format |
Tensor.index_fill | 支持 | Ascend上暂不支持 |
Tensor.inverse | 支持 | 不支持Ascend |
Tensor.is_floating_point | 支持 | |
Tensor.norm | 支持 | |
Tensor.real | 支持 | Ascend上暂不支持 |
Tensor.scatter_ | 支持 | |
Tensor.scatter | 支持 | 不支持reduce='mutiply', Ascend不支持reduce='add', 不支持indices.shape != src.shape |
Tensor.std | 支持 | 不支持复数和float64输入 |
Tensor.svd | 支持 | Ascend上暂不支持 |
Tensor.to | 支持 | |
Tensor.unique | 支持 | |
Tensor.view | 支持 | |
Tensor.where | 支持 | |
Tensor.xlogy | 支持 | Ascend 上当前只支持float16 和float32输入 |
Tensor.abs_ | 部分支持 | inplace不支持图模式 |
Tensor.absolute_ | 部分支持 | inplace不支持图模式 |
Tensor.acos_ | 部分支持 | inplace不支持图模式 |
Tensor.arccos_ | 部分支持 | inplace不支持图模式 |
Tensor.addr_ | 部分支持 | inplace不支持图模式 |
Tensor.add_ | 部分支持 | inplace不支持图模式 |
Tensor.addbmm_ | 部分支持 | inplace不支持图模式 |
Tensor.addcdiv_ | 部分支持 | inplace不支持图模式 |
Tensor.addcmul_ | 部分支持 | inplace不支持图模式 |
Tensor.addmm_ | 部分支持 | inplace不支持图模式 |
Tensor.addmv_ | 部分支持 | inplace不支持图模式 |
Tensor.addr_ | 部分支持 | inplace不支持图模式 |
Tensor.asin_ | 部分支持 | inplace不支持图模式 |
Tensor.arcsin_ | 部分支持 | inplace不支持图模式 |
Tensor.atan_ | 部分支持 | inplace不支持图模式 |
Tensor.arctan_ | 部分支持 | inplace不支持图模式 |
Tensor.atan2_ | 部分支持 | inplace不支持图模式 |
Tensor.arctan2_ | 部分支持 | inplace不支持图模式 |
Tensor.baddbmm_ | 部分支持 | inplace不支持图模式 |
Tensor.bitwise_not_ | 部分支持 | inplace不支持图模式 |
Tensor.bitwise_and_ | 部分支持 | inplace不支持图模式 |
Tensor.bitwise_or_ | 部分支持 | inplace不支持图模式 |
Tensor.bitwise_xor_ | 部分支持 | inplace不支持图模式 |
Tensor.clamp_ | 部分支持 | inplace不支持图模式 |
Tensor.clip_ | 部分支持 | inplace不支持图模式 |
Tensor.copy_ | 部分支持 | inplace不支持图模式 |
Tensor.copysign_ | 部分支持 | inplace不支持图模式 |
Tensor.acosh_ | 部分支持 | inplace不支持图模式 |
Tensor.arccosh_ | 部分支持 | inplace不支持图模式 |
Tensor.cumprod_ | 部分支持 | inplace不支持图模式 |
Tensor.div_ | 部分支持 | inplace不支持图模式 |
Tensor.divide_ | 部分支持 | inplace不支持图模式 |
Tensor.eq_ | 部分支持 | inplace不支持图模式 |
Tensor.expm1_ | 部分支持 | inplace不支持图模式 |
Tensor.fix_ | 部分支持 | inplace不支持图模式 |
Tensor.fill_ | 部分支持 | inplace不支持图模式 |
Tensor.float_power_ | 部分支持 | inplace不支持图模式 |
Tensor.fmod_ | 部分支持 | inplace不支持图模式 |
Tensor.ge_ | 部分支持 | inplace不支持图模式 |
Tensor.greater_equal_ | 部分支持 | inplace不支持图模式 |
Tensor.gt_ | 部分支持 | inplace不支持图模式 |
Tensor.greater_ | 部分支持 | inplace不支持图模式 |
Tensor.hypot_ | 部分支持 | inplace不支持图模式 |
Tensor.le_ | 部分支持 | inplace不支持图模式 |
Tensor.less_equal_ | 部分支持 | inplace不支持图模式 |
Tensor.lt_ | 部分支持 | inplace不支持图模式 |
Tensor.less_ | 部分支持 | inplace不支持图模式 |
Tensor.mul_ | 部分支持 | inplace不支持图模式 |
Tensor.multiply_ | 部分支持 | inplace不支持图模式 |
Tensor.ne_ | 部分支持 | inplace不支持图模式 |
Tensor.not_equal_ | 部分支持 | inplace不支持图模式 |
Tensor.neg_ | 部分支持 | inplace不支持图模式 |
Tensor.negative_ | 部分支持 | inplace不支持图模式 |
Tensor.pow_ | 部分支持 | inplace不支持图模式 |
Tensor.reciprocal_ | 部分支持 | inplace不支持图模式 |
Tensor.renorm | 部分支持 | inplace不支持图模式 |
Tensor.resize_ | 部分支持 | inplace不支持图模式 |
Tensor.sigmoid_ | 部分支持 | inplace不支持图模式 |
Tensor.sign_ | 部分支持 | inplace不支持图模式 |
Tensor.sin_ | 部分支持 | inplace不支持图模式 |
Tensor.sinc_ | 部分支持 | inplace不支持图模式 |
Tensor.sinh_ | 部分支持 | inplace不支持图模式 |
Tensor.asinh_ | 部分支持 | inplace不支持图模式 |
Tensor.square_ | 部分支持 | inplace不支持图模式 |
Tensor.sqrt_ | 部分支持 | inplace不支持图模式 |
Tensor.squeeze_ | 部分支持 | inplace不支持图模式 |
Tensor.sub_ | 部分支持 | inplace不支持图模式 |
Tensor.tan_ | 部分支持 | inplace不支持图模式 |
Tensor.tanh_ | 部分支持 | inplace不支持图模式 |
Tensor.atanh_ | 部分支持 | inplace不支持图模式 |
Tensor.arctanh_ | 部分支持 | inplace不支持图模式 |
Tensor.transpose_ | 部分支持 | inplace不支持图模式 |
Tensor.trunc_ | 部分支持 | inplace不支持图模式 |
Tensor.unsqueeze_ | 部分支持 | inplace不支持图模式 |
Tensor.zero_ | 部分支持 | inplace不支持图模式 |
MSAdapter接口 | 状态 | 备注 |
---|---|---|
nn.ModuleDict | 支持 | |
nn.ParameterList | 支持 | |
nn.ParameterDict | 支持 | |
nn.Unfold | 支持 | |
nn.Fold | 支持 | |
nn.MaxPool1d | 支持 | 不支持Ascend |
nn.MaxPool2d | 支持 | |
nn.MaxPool3d | 支持 | |
nn.AvgPool1d | 支持 | 不支持参数ceilmode,countincludepad |
nn.AvgPool2d | 支持 | 不支持参数ceilmode,countincludepad |
nn.AvgPool3d | 支持 | 不支持参数ceilmode,countincludepad |
nn.FractionalMaxPool2d | 支持 | |
nn.FractionalMaxPool3d | 支持 | |
nn.LPPool1d | 支持 | 不支持float64 |
nn.LPPool2d | 支持 | 不支持3维,kernelsize不支持tuple, 不支持float64 |
nn.AdaptiveMaxPool1d | 支持 | |
nn.AdaptiveMaxPool2d | 支持 | |
nn.AdaptiveMaxPool3d | 支持 | 不支持float64 |
nn.AdaptiveAvgPool1d | 支持 | 不支持2维 |
nn.AdaptiveAvgPool2d | 支持 | 不支持3维 |
nn.AdaptiveAvgPool3d | 支持 | 不支持4维,不支持float64 |
nn.ReflectionPad1d | 支持 | |
nn.ReflectionPad2d | 支持 | 不支持complex32 |
nn.ReflectionPad3d | 支持 | |
nn.ReplicationPad1d | 支持 | |
nn.ReplicationPad2d | 支持 | 不支持3维 |
nn.ReplicationPad3d | 支持 | |
nn.ZeroPad2d | 支持 | |
nn.ConstantPad1d | 支持 | |
nn.ConstantPad2d | 支持 | |
nn.ConstantPad3d | 支持 | |
nn.ELU | 支持 | Alpha仅支持1.0 |
nn.Hardshrink | 支持 | 不支持float64 |
nn.Hardsigmoid | 支持 | |
nn.Hardtanh | 支持 | 不支持float64 |
nn.Hardswish | 支持 | 不支持float64 |
nn.LeakyReLU | 支持 | 不支持float64 |
nn.LogSigmoid | 支持 | |
nn.PReLU | 支持 | 不支持float64 |
nn.ReLU | 支持 | |
nn.ReLU6 | 支持 | 不支持float64 |
nn.RReLU | 支持 | inplace不支持GRAPH模式 |
nn.SELU | 支持 | inplace不支持GRAPH模式 |
nn.CELU | 支持 | inplace不支持GRAPH模式 |
nn.GELU | 支持 | |
nn.Sigmoid | 支持 | |
nn.SiLU | 支持 | |
nn.Mish | 支持 | inplace不支持GRAPH模式 |
nn.Softplus | 支持 | |
nn.Softshrink | 支持 | |
nn.Softsign | 支持 | |
nn.Tanh | 支持 | |
nn.Tanhshrink | 支持 | |
nn.Threshold | 支持 | inplace不支持GRAPH模式 |
nn.GLU | 支持 | |
nn.Softmin | 支持 | |
nn.Softmax | 支持 | |
nn.Softmax2d | 支持 | |
nn.LogSoftmax | 支持 | 不支持float64,不支持8维及以上 |
nn.BatchNorm1d | 支持 | 不支持3维 |
nn.BatchNorm2d | 支持 | |
nn.BatchNorm3d | 支持 | |
nn.LazyBatchNorm1d | 支持 | |
nn.LazyBatchNorm2d | 支持 | |
nn.LazyBatchNorm3d | 支持 | |
nn.GroupNorm | 支持 | |
nn.LayerNorm | 支持 | |
nn.LocalResponseNorm | 支持 | |
nn.RNNBase | 支持 | |
nn.RNN | 支持 | |
nn.RNNCell | 支持 | |
nn.LSTMCell | 支持 | |
nn.GRUCell | 支持 | |
nn.Identity | 支持 | |
nn.Linear | 支持 | device, dtype参数不支持 |
nn.Bilinear | 支持 | |
nn.LazyLinear | 支持 | |
nn.Dropout | 支持 | |
nn.Dropout1d | 支持 | |
nn.Dropout2d | 支持 | |
nn.Dropout3d | 支持 | |
nn.AlphaDropout | 支持 | |
nn.FeatureAlphaDropout | 支持 | |
nn.CosineSimilarity | 支持 | |
nn.PairwiseDistance | 支持 | |
nn.L1Loss | 支持 | |
nn.MSELoss | 支持 | |
nn.CrossEntropyLoss | 支持 | |
nn.CTCLoss | 支持 | |
nn.NLLLoss | 支持 | |
nn.PoissonNLLLoss | 支持 | |
nn.GaussianNLLLoss | 支持 | |
nn.BCELoss | 支持 | |
nn.BCEWithLogitsLoss | 支持 | |
nn.MarginRankingLoss | 支持 | |
nn.HingeEmbeddingLoss | 支持 | 不支持int输入 |
nn.HuberLoss | 支持 | |
nn.SmoothL1Loss | 支持 | |
nn.SoftMarginLoss | 支持 | |
nn.MultiLabelSoftMarginLoss | 支持 | |
nn.CosineEmbeddingLoss | 支持 | |
nn.TripletMarginWithDistanceLoss | 支持 | |
nn.PixelShuffle | 支持 | |
nn.PixelUnshuffle | 支持 | |
nn.Upsample | 支持 | |
nn.UpsamplingNearest2d | 支持 | 不支持size为none |
nn.UpsamplingBilinear2d | 支持 | |
nn.ChannelShuffle | 支持 | |
nn.Flatten | 支持 | |
nn.Unflatten | 支持 | |
nn.Module | 支持 | |
nn.Sequential | 支持 | |
nn.ModuleList | 支持 | |
nn.Conv1d | 支持 | 1.不支持2维;2.pad类型支持不完备;3.部分参数不支持tuple |
nn.Conv2d | 支持 | 不支持3维,pad类型支持不完备 |
nn.Conv3d | 支持 | 不支持复数 |
nn.ConvTranspose1d | 支持 | 1.不支持2维;2.pad类型支持不完备;3.部分参数不支持tuple |
nn.ConvTranspose2d | 支持 | 1.不支持3维输入;2.pad类型支持不完备 |
nn.ConvTranspose3d | 支持 | 1.不支持4维输入;2.pad类型支持不完备 |
nn.LazyConv1d | 支持 | pad类型支持不完备 |
nn.LazyConv2d | 支持 | pad类型支持不完备 |
nn.LazyConv3d | 支持 | pad类型支持不完备 |
nn.LazyConvTranspose1d | 支持 | pad类型支持不完备 |
nn.LazyConvTranspose2d | 支持 | pad类型支持不完备 |
nn.LazyConvTranspose3d | 支持 | pad类型支持不完备 |
nn.MaxUnpool1d | 支持 | Ascend上不支持 |
nn.MaxUnpool2d | 支持 | Ascend上不支持 |
nn.MaxUnpool3d | 支持 | Ascend上不支持 |
nn.MultiheadAttention | 支持 | |
nn.AdaptiveLogSoftmaxWithLoss | 支持 | 不支持GRAPH模式 |
nn.SyncBatchNorm | 支持 | 仅支持Ascend |
nn.InstanceNorm1d | 支持 | 仅支持GPU |
nn.InstanceNorm2d | 支持 | 仅支持GPU |
nn.InstanceNorm3d | 支持 | 仅支持GPU |
nn.LazyInstanceNorm1d | 支持 | 仅支持GPU |
nn.LazyInstanceNorm2d | 支持 | 仅支持GPU |
nn.LazyInstanceNorm3d | 支持 | 仅支持GPU |
nn.LSTM | 支持 | 当前proj_size不支持 |
nn.GRU | 支持 | |
nn.Embedding | 支持 | |
nn.KLDivLoss | 支持 | cpu不支持 |
nn.MultiLabelMarginLoss | 支持 | 不支持cpu |
nn.MultiMarginLoss | 支持 | Ascend上暂不支持 |
nn.Module.named_module | 支持 | |
nn.TripletMarginLoss | 支持 | Ascend上暂不支持;不支持1维 |
MSAdapter接口 | 状态 | 备注 |
---|---|---|
functional.max_pool2d | 支持 | |
functional.max_pool3d | 支持 | |
functional.conv_transpose2d | 支持 | |
functional.conv_transpose3d | 支持 | |
functional.avg_pool2d | 支持 | 不支持3维,后三个参数不支持 |
functional.avg_pool3d | 支持 | |
functional.max_pool1d | 支持 | |
functional.max_unpool1d | 支持 | |
functional.max_unpool2d | 支持 | |
functional.max_unpool3d | 支持 | |
functional.lp_pool1d | 支持 | 不支持float64 |
functional.lp_pool2d | 支持 | 不支持float64 |
functional.adaptive_max_pool1d | 支持 | |
functional.adaptive_max_pool2d | 支持 | |
functional.adaptive_avg_pool1d | 支持 | |
functional.fractional_max_pool2d | 支持 | |
functional.fractional_max_pool3d | 支持 | |
functional.threshold | 支持 | |
functional.threshold_ | 支持 | |
functional.relu | 支持 | |
functional.relu_ | 支持 | |
functional.hardtanh | 支持 | |
functional.hardtanh_ | 支持 | |
functional.hardswish | 支持 | |
functional.relu6 | 支持 | |
functional.elu | 支持 | |
functional.elu_ | 支持 | |
functional.selu | 支持 | |
functional.celu | 支持 | |
functional.leaky_relu | 支持 | |
functional.leaky_relu_ | 支持 | |
functional.prelu | 支持 | inplace不支持GRAPH模式 |
functional.rrelu | 支持 | inplace不支持GRAPH模式 |
functional.rrelu_ | 支持 | |
functional.glu | 支持 | |
functional.gelu | 支持 | |
functional.logsigmoid | 支持 | |
functional.hardshrink | 支持 | |
functional.tanhshrink | 支持 | |
functional.softsign | 支持 | |
functional.softplus | 支持 | |
functional.softmin | 支持 | |
functional.softmax | 支持 | 不支持_stacklevel |
functional.softshrink | 支持 | |
functional.gumbel_softmax | 支持 | |
functional.log_softmax | 支持 | 不支持float64,不支持_stacklevel |
functional.tanh | 支持 | |
functional.sigmoid | 支持 | |
functional.hardsigmoid | 支持 | |
functional.silu | 支持 | |
functional.mish | 支持 | |
functional.batch_norm | 支持 | |
functional.group_norm | 支持 | |
functional.instance_norm | 支持 | |
functional.layer_norm | 支持 | |
functional.local_response_norm | 支持 | |
functional.normalize | 支持 | |
functional.linear | 支持 | |
functional.bilinear | 支持 | |
functional.dropout | 支持 | |
functional.alpha_dropout | 支持 | |
functional.feature_alpha_dropout | 支持 | |
functional.dropout1d | 支持 | inplace不支持GRAPH模式 |
functional.dropout2d | 支持 | inplace不支持GRAPH模式 |
functional.dropout3d | 支持 | inplace不支持GRAPH模式 |
functional.one_hot | 支持 | |
functional.pairwise_distance | 支持 | |
functional.cosine_similarity | 支持 | |
functional.pdist | 支持 | |
functional.binary_cross_entropy | 支持 | |
functional.binary_cross_entropy_with_logits | 支持 | |
functional.poisson_nll_loss | 支持 | |
functional.cosine_embedding_loss | 支持 | |
functional.cross_entropy | 支持 | |
functional.gaussian_nll_loss | 支持 | |
functional.hinge_embedding_loss | 支持 | |
functional.l1_loss | 支持 | |
functional.mse_loss | 支持 | |
functional.margin_ranking_loss | 支持 | |
functional.multilabel_soft_margin_loss | 支持 | |
functional.nll_loss | 支持 | |
functional.huber_loss | 支持 | |
functional.smooth_l1_loss | 支持 | |
functional.soft_margin_loss | 支持 | |
functional.triplet_margin_loss | 支持 | |
functional.triplet_margin_with_distance_loss | 支持 | |
functional.pixel_shuffle | 支持 | |
functional.pixel_unshuffle | 支持 | |
functional.grid_sample | 支持 | |
functional.huber_loss | 支持 | |
functional.multi_margin_loss | 支持 | |
functional.triplet_margin_loss | 支持 | |
functional.conv1d | 支持 | |
functional.conv2d | 支持 | |
functional.conv3d | 支持 | |
functional.conv_transpose1d | 支持 | |
functional.unfold | 支持 | 不支持GPU |
functional.fold | 支持 | 不支持Ascend |
functional.adaptive_max_pool3d | 支持 | 不支持Ascend |
functional.adaptive_avg_pool2d | 支持 | |
functional.adaptive_avg_pool3d | 支持 | |
functional.embedding | 支持 | |
functional.ctc_loss | 支持 | |
functional.kl_div | 支持 | |
functional.multilabel_margin_loss | 支持 | 不支持CPU |
functional.multi_margin_loss | 支持 | |
functional.interpolate | 支持 | |
functional.upsample | 支持 | |
functional.upsample_nearest | 支持 | |
functional.upsample_bilinear | 支持 | |
functional.affine_grid | 支持 | |
functional.max_pool2d | 支持 |
MSAdapter接口 | 状态 | 备注 |
---|---|---|
norm | 不支持 | |
vector_norm | 不支持 | |
matrix_norm | 不支持 | |
diagonal | 支持 | |
det | 不支持 | |
slogdet | 不支持 | |
cond | 不支持 | |
matrix_rank | 不支持 | |
cholesky | 不支持 | |
qr | 不支持 | |
lu | 不支持 | |
lu_factor | 不支持 | |
eig | 不支持 | |
eigvals | 不支持 | |
eigh | 支持 | |
eigvalsh | 不支持 | |
svd | 不支持 | |
svdvals | 不支持 | |
solve | 支持 | |
solve_triangular | 不支持 | |
lu_solve | 不支持 | |
lstsq | 不支持 | |
inv | 不支持 | |
pinv | 不支持 | |
matrix_exp | 不支持 | |
matrix_power | 不支持 | |
cross | 不支持 | |
matmul | 不支持 | |
vecdot | 不支持 | |
multi_dot | 不支持 | |
householder_product | 不支持 | |
tensorinv | 不支持 | |
tensorsolve | 不支持 | |
vander | 不支持 | |
cholesky_ex | 不支持 | |
inv_ex | 不支持 | |
solve_ex | 不支持 | |
lu_factor_ex | 不支持 | |
ldl_factor | 不支持 | |
ldl_factor_ex | 不支持 | |
ldl_solve | 不支持 | |
eigh | 支持 | |
solve | 支持 |
Dear OpenI User
Thank you for your continuous support to the Openl Qizhi Community AI Collaboration Platform. In order to protect your usage rights and ensure network security, we updated the Openl Qizhi Community AI Collaboration Platform Usage Agreement in January 2024. The updated agreement specifies that users are prohibited from using intranet penetration tools. After you click "Agree and continue", you can continue to use our services. Thank you for your cooperation and understanding.
For more agreement content, please refer to the《Openl Qizhi Community AI Collaboration Platform Usage Agreement》