torch

Tensor

API名称	API状态	限制与说明
torch.is_tensor	Not Support	N/A
torch.is_storage	Not Support	N/A
torch.is_complex	Not Support	N/A
torch.is_conj	Not Support	N/A
torch.is_floating_point	Beta	支持数据类型：fp32
torch.is_nonzero	Not Support	N/A
torch.set_default_dtype	Not Support	N/A
torch.get_default_dtype	Not Support	N/A
torch.set_default_device	Not Support	N/A
torch.get_default_device	Not Support	N/A
torch.set_default_tensor_type	Not Support	N/A
torch.numel	Beta	N/A
torch.set_printoptions	Not Support	N/A
torch.set_flush_denormal	Not Support	N/A

Creation Ops

API名称	API状态	限制与说明
torch.tensor	Not Support	N/A
torch.sparse_coo_tensor	Not Support	N/A
torch.sparse_csr_tensor	Not Support	N/A
torch.sparse_csc_tensor	Not Support	N/A
torch.sparse_bsr_tensor	Not Support	N/A
torch.sparse_bsc_tensor	Not Support	N/A
torch.asarray	Not Support	N/A
torch.as_tensor	Not Support	N/A
torch.as_strided	Beta	支持数据类型：fp32
torch.from_file	Not Support	N/A
torch.from_numpy	Not Support	N/A
torch.from_dlpack	Not Support	N/A
torch.frombuffer	Not Support	N/A
torch.zeros	Beta	不支持out、layout参数；requires_grad参数可以传入但不生效
torch.zeros_like	Beta	不支持layout、device、requires_grad参数；支持数据类型：bf16、fp16、fp32、uint8、int8、int16、int32、int64、bool
torch.ones	Beta	不支持out、layout、requires_grad参数；device参数可以传入但不生效；支持数据类型：bf16、fp16、fp32、fp64、uint8、int8、int16、int32、int64、bool
torch.ones_like	Beta	不支持layout、requires_grad、memory_format参数；支持数据类型：bf16、fp16、fp32、fp64、uint8、int8、int16、int32、int64、bool
torch.arange	Beta	不支持out、layout、requires_grad参数；end参数具有默认值None，torch无默认值；可传入device参数，但不会生效； 支持数据类型：bf16、fp16、fp32、fp64、int32、int64
torch.range	Beta	不支持out、requires_grad参数
torch.linspace	Beta	不支持out、layout、device、requires_grad参数；支持数据类型：bf16、fp16、fp32、fp64
torch.logspace	Beta	不支持out出参
torch.eye	Beta	不支持out、layout、device、requires_grad参数；支持数据类型：fp16、fp32
torch.empty	Beta	不支持out出参、pin_memory、memory_format参数；支持数据类型：bf16、fp16、fp32、fp64、uint8、int8、int16、int32、int64、bool
torch.empty_like	Beta	支持数据类型：bf16、uint8、int8、int16、int32、int64、bool
torch.empty_strided	Not Support	N/A
torch.full	Beta	可传入device参数，但不会生效； 不支持out、layout、device、require_grad参数；支持数据类型：fp32
torch.full_like	Beta	不支持out、layout、device、require_grad参数；支持数据类型：uint8、int8、int16、int32、int64、bool
torch.quantize_per_tensor	Not Support	N/A
torch.quantize_per_channel	Not Support	N/A
torch.dequantize	Not Support	N/A
torch.complex	Not Support	N/A
torch.polar	Beta	不支持out出参；支持数据类型：fp32
torch.heaviside	Beta	不支持out出参
torch.linalg.norm	Demo	N/A
torch.linalg.vector_norm	Demo	不支持CPU平台

Indexing, Slicing, Joining, Mutation Ops

API名称	API状态	限制与说明
torch.adjoint	Not Support	N/A
torch.argwhere	Beta	支持数据类型：fp16、fp32、uint8、int8、int16、int32、int64、bool
torch.cat	Stable	支持数据类型：bf16、fp16、fp32、fp64、uint8、int8、int16、int32、int64、bool
torch.concat	Stable	支持数据类型：bf16、fp16、fp32、uint8、int8、int16、int32、int64、bool
torch.concatenate	Stable	支持数据类型：bf16、fp16、fp32、int64、bool
torch.conj	Beta	支持数据类型：bf16、fp16、fp32、uint8、int8、int16、int32、int64、bool
torch.chunk	Stable	支持数据类型：bf16、fp16、fp32、fp64、uint8、int8、int16、int32、int64、bool
torch.dsplit	Not Support	N/A
torch.column_stack	Not Support	N/A
torch.dstack	Not Support	N/A
torch.gather	Beta	不支持out、sparse_grad参数；支持数据类型：fp16、fp32、int16、int32、int64、bool
torch.hsplit	Not Support	N/A
torch.hstack	Beta	不支持out出参；支持数据类型：bf16、fp16、fp32、uint8、int8、int16、int32、int64、bool
torch.index_add	Beta	不支持out出参；支持数据类型：fp16、fp32、int64、bool
torch.index_copy	Not Support	N/A
torch.index_reduce	Not Support	N/A
torch.index_select	Stable	支持数据类型：bf16、fp16、fp32、int16、int32、int64、bool
torch.masked_select	Stable	支持数据类型：fp16、fp32、int16、int32、int64、bool
torch.movedim	Not Support	N/A
torch.moveaxis	Beta	支持数据类型：int64、float
torch.narrow	Stable	支持数据类型：bf16、fp16、fp32、fp64、uint8、int8、int16、int32、int64、bool
torch.narrow_copy	Not Support	N/A
torch.nonzero	Beta	不支持out出参；支持数据类型：fp16、fp32、uint8、int8、int16、int32、int64、bool
torch.permute	Stable	支持数据类型：bf16、fp16、fp32、uint8、int8、int16、int32、int64、bool
torch.reshape	Stable	支持数据类型：bf16、fp16、fp32、fp64、uint8、int8、int16、int32、int64、bool
torch.row_stack	Not Support	N/A
torch.select	Stable	N/A
torch.scatter	Stable	支持数据类型：bf16、fp16、fp32、fp64、uint8、int8、int16、int32、int64、bool
torch.diagonal_scatter	Not Support	N/A
torch.select_scatter	Not Support	N/A
torch.slice_scatter	Not Support	N/A
torch.scatter_add	Stable	N/A
torch.scatter_reduce	Not Support	N/A
torch.split	Stable	支持数据类型：bf16、fp16、fp32、uint8、int8、int16、int32、int64、bool
torch.squeeze	Beta	支持数据类型：bf16、fp16、fp32、fp64、uint8、int8、int16、int32、int64、bool
torch.stack	Stable	支持数据类型：bf16、fp16、fp32、fp64、uint8、int8、int16、int32、int64、bool
torch.swapaxes	Beta	支持数据类型：bf16、fp16、fp32、fp64、uint8、int8、int16、int32、int64、bool
torch.swapdims	Beta	支持数据类型：bf16、fp16、fp32、fp64、uint8、int8、int16、int32、int64、bool
torch.t	Not Support	N/A
torch.take	Beta	支持数据类型：fp16、fp32、int16、int32、int64、bool
torch.take_along_dim	Not Support	N/A
torch.tensor_split	Beta	支持数据类型：bf16、fp16、fp32、uint8、int8、int16、int32、int64、bool
torch.tile	Stable	支持数据类型：bf16、fp16、fp32、fp64、uint8、int8、int16、int32、int64、bool
torch.transpose	Stable	支持数据类型：bf16、fp16、fp32、uint8、int8、int16、int32、int64、bool
torch.unbind	Beta	N/A
torch.unravel_index	Not Support	N/A
torch.unsqueeze	Stable	支持数据类型：bf16、fp16、fp32、fp64、uint8、int8、int16、int32、int64、bool
torch.vsplit	Not Support	N/A
torch.vstack	Beta	不支持out出参；支持数据类型：bf16、fp16、fp32、fp64、uint8、int8、int16、int32、int64、bool
torch.where	Stable	bf16、fp16、fp32、fp64、uint8、int8、int16、int32、int64、bool；不支持8维度的shape

Accelerators

API名称	API状态	限制与说明
torch.Stream	Not Support	N/A
torch.Event	Not Support	N/A

Generators

API名称	API状态	限制与说明
torch.Generator	Not Support	N/A

Random sampling

API名称	API状态	限制与说明
torch.seed	Not Support	N/A
torch.manual_seed	Not Support	N/A
torch.initial_seed	Not Support	N/A
torch.get_rng_state	Not Support	N/A
torch.set_rng_state	Not Support	N/A
torch.bernoulli	Stable	支持数据类型：fp32
torch.multinomial	Beta	不支持out出参；支持数据类型：fp16、fp32
torch.normal	Beta	mean参数有默认值0.0，torch没有；std参数有默认值1.0，torch没有；不支持generator、out参数；支持数据类型：fp16、fp32
torch.poisson	Not Support	N/A
torch.rand	Beta	不支持generator、out、layout、requires_grad参数
torch.rand_like	Beta	不支持layout、device、requires_grad、memory_format参数；支持数据类型：支持bf16、fp16、fp32、fp64、uint8、int8、int16、int32、int64
torch.randint	Stable	N/A
torch.randint_like	Not Support	N/A
torch.randn	Beta	参数out、layout、device、requires_grad、pin_memory可以传入，但是这些参数不生效
torch.randn_like	Beta	不支持layout、device、requires_grad、memory_format参数；支持数据类型：fp32
torch.randperm	Beta	不支持out、layout、device、requires_grad、pin_memory参数

In-place random sampling

暂不支持

Quasi-random sampling

API名称	API状态	限制与说明
torch.quasirandom.SobolEngine	Not Support	N/A

Serialization

API名称	API状态	限制与说明
torch.save	Not Support	N/A
torch.load	Not Support	N/A

Parallelism

API名称	API状态	限制与说明
torch.get_num_threads	Not Support	N/A
torch.set_num_threads	Not Support	N/A
torch.get_num_interop_threads	Not Support	N/A
torch.set_num_interop_threads	Not Support	N/A

Locally disabling gradient computation

API名称	API状态	限制与说明
torch.no_grad	Not Support	N/A
torch.enable_grad	Not Support	N/A
torch.set_grad_enabled	Not Support	N/A
torch.is_grad_enabled	Not Support	N/A
torch.inference_mode	Not Support	N/A
torch.is_inference_mode_enabled	Not Support	N/A

Math operations

Constants

API名称	API状态	限制与说明
torch.inf	Not Support	N/A
torch.nan	Not Support	N/A

Pointwise Ops

API名称	API状态	限制与说明
torch.abs	Stable	支持数据类型：bf16、fp16、fp32、fp64、uint8、int8、int16、int32、int64、bool
torch.absolute	Stable	支持数据类型：bf16、fp16、fp32、uint8、int8、int16、int32、int64
torch.acos	Stable	支持数据类型：fp16、fp32、uint8、int8、int16、int32、int64、bool
torch.arccos	Not Support	N/A
torch.acosh	Stable	支持数据类型：bf16、fp16、fp32、fp64、uint8、int8、int16、int32、int64、bool
torch.arccosh	Beta	不支持out出参；支持数据类型：bf16、fp16、fp32、fp64、uint8、int8、int16、int32、int64、bool
torch.add	Stable	支持数据类型：bf16、fp16、fp32、fp64、uint8、int8、int16、int32、int64、bool
torch.addcdiv	Beta	不支持out出参；支持数据类型：fp16、fp32、int64
torch.addcmul	Beta	不支持out出参；支持数据类型：bf16、fp16、fp32、fp64、uint8、int8、int32、int64
torch.angle	Beta	不支持out出参；支持数据类型：fp16、fp32、uint8、int8、int16、int32、int64、bool
torch.asin	Stable	支持数据类型：bf16、fp16、fp32、uint8、int8、int16、int32、int64、bool
torch.arcsin	Stable	支持数据类型：bf16、fp16、fp32、uint8、int8、int16、int32、int64、bool
torch.asinh	Stable	支持数据类型：fp16、fp32、uint8、int8、int16、int32、int64、bool
torch.arcsinh	Stable	支持数据类型：fp16、fp32、uint8、int8、int16、int32、int64、bool
torch.atan	Stable	支持数据类型：fp16、fp32、uint8、int8、int16、int32、int64、bool
torch.arctan	Stable	支持数据类型：fp16、fp32、uint8、int8、int16、int32、int64、bool
torch.atanh	Stable	支持数据类型：fp16、fp32、uint8、int8、int16、int32、int64、bool
torch.arctanh	Stable	支持数据类型：fp16、fp32、uint8、int8、int16、int32、int64、bool
torch.atan2	Stable	支持数据类型：fp16、fp32、uint8、int8、int16、int32、int64、bool
torch.arctan2	Stable	支持数据类型：fp16、fp32、uint8、int8、int16、int32、int64、bool
torch.bitwise_not	Not Support	N/A
torch.bitwise_and	Stable	支持数据类型：uint8、int8、int16、int32、int64、bool
torch.bitwise_or	Stable	支持数据类型：uint8、int8、int16、int32、int64、bool
torch.bitwise_xor	Stable	支持数据类型：uint8、int8、int16、int32、int64、bool
torch.bitwise_left_shift	Beta	不支持out出参
torch.bitwise_right_shift	Beta	不支持out出参
torch.ceil	Stable	支持数据类型：fp16、fp32
torch.clamp	Stable	支持数据类型：bf16、fp16、fp32、uint8、int8、int16、int32、int64
torch.clip	Beta	不支持out出参；支持数据类型：bf16、fp16、fp32、uint8、int8、int16、int32、int64
torch.conj_physical	Not Support	N/A
torch.copysign	Not Support	N/A
torch.cos	Stable	支持数据类型：bf16、fp16、fp32、uint8、int8、int16、int32、int64、bool
torch.cosh	Stable	支持数据类型：bf16、fp16、fp32、fp64、uint8、int8、int16、int32、int64、bool
torch.deg2rad	Beta	不支持out出参；支持数据类型：bf16、fp16、fp32、fp64、uint8、int8、int16、int32、int64、bool
torch.div	Stable	支持数据类型：bf16、fp16、fp32、uint8、int8、int16、int32、int64、bool
torch.divide	Beta	不支持out出参；支持数据类型：bf16、fp16、fp32、uint8、int8、int16、int32、int64、bool
torch.digamma	Beta	不支持out出参
torch.erf	Stable	支持数据类型：fp16、fp32、int64、bool
torch.erfc	Stable	支持数据类型：fp16、fp32、int64、bool
torch.erfinv	Stable	支持数据类型：fp16、fp32、uint8、int8、int16、int32、int64、bool
torch.exp	Stable	支持数据类型：bf16、fp16、fp32、fp64、int64、bool
torch.exp2	Beta	不支持out出参；支持数据类型：bf16、fp16、fp32、uint8、int8、int16、int32、int64、bool
torch.expm1	Stable	支持数据类型：fp16、fp32、int64、bool
torch.fake_quantize_per_channel_affine	Not Support	N/A
torch.fake_quantize_per_tensor_affine	Not Support	N/A
torch.fix	Not Support	N/A
torch.float_power	Beta	不支持out出参；支持数据类型：bf16、fp16、fp32、uint8、int8、int16、int32、int64、bool
torch.floor	Stable	支持数据类型：fp16、fp32
torch.floor_divide	Beta	不支持out出参；支持数据类型：fp16、fp32、uint8、int8、int16、int32、int64
torch.fmod	Beta	不支持out出参；支持数据类型：fp16、fp32、uint8、int8、int32、int64
torch.frac	Beta	不支持out出参
torch.frexp	Not Support	N/A
torch.gradient	Not Support	N/A
torch.imag	Beta	N/A
torch.ldexp	Not Support	N/A
torch.lerp	Beta	不支持out出参；支持数据类型：支持fp16、fp32
torch.lgamma	Beta	不支持out出参
torch.log	Stable	支持数据类型：bf16、fp16、fp32、fp64、uint8、int8、int16、int32、int64、bool
torch.log10	Beta	支持数据类型：bf16、fp16、fp32、fp64、uint8、int8、int16、int32、int64、bool
torch.log1p	Stable	支持数据类型：fp16、fp32、uint8、int8、int16、int32、int64、bool
torch.log2	Stable	支持数据类型：bf16、fp32、int64、bool、fp16
torch.logaddexp	Not Support	N/A
torch.logaddexp2	Not Support	N/A
torch.logical_and	Stable	支持数据类型：bf16、fp16、fp32、uint8、int8、int16、int32、int64、bool
torch.logical_not	Stable	支持数据类型：fp16、fp32、uint8、int8、int16、int32、int64、bool
torch.logical_or	Stable	支持数据类型：bf16、fp16、fp32、uint8、int8、int16、int32、int64、bool
torch.logical_xor	Stable	支持数据类型：fp16、fp32、uint8、int8、int16、int32、int64、bool
torch.logit	Beta	不支持out出参
torch.hypot	Beta	不支持out出参
torch.i0	Not Support	N/A
torch.igamma	Beta	不支持out出参
torch.igammac	Beta	不支持out出参
torch.mul	Stable	支持数据类型：bf16、fp16、fp32、uint8、int8、int16、int32、int64、bool
torch.multiply	Not Support	N/A
torch.mvlgamma	Beta	不支持out出参
torch.nan_to_num	Stable	支持数据类型：bf16、fp16、fp32、uint8、int8、int16、int32、int64、bool
torch.nan_to_num_	Demo	支持数据类型：bf16、fp16、fp32、uint8、int8、int16、int32、int64、bool
torch.neg	Stable	支持数据类型：bf16、fp16、fp32、int8、int32、int64
torch.negative	Beta	不支持out出参；支持数据类型：bf16、fp16、fp32、int8、int32、int64
torch.nextafter	Beta	不支持out出参
torch.polygamma	Beta	不支持out出参
torch.positive	Beta	支持数据类型：bf16、fp16、fp32、uint8、int8、int16、int32、int64
torch.pow	Stable	支持数据类型：bf16、fp16、fp32、fp64、int16、int32、int64
torch.quantized_batch_norm	Not Support	N/A
torch.quantized_max_pool1d	Not Support	N/A
torch.quantized_max_pool2d	Not Support	N/A
torch.rad2deg	Beta	不支持out出参；支持数据类型：bf16、fp16、fp32、fp64、uint8、int8、int16、int32、int64、bool
torch.real	Beta	支持数据类型：fp16、fp32
torch.reciprocal	Stable	支持数据类型：bf16、fp16、fp32、uint8、int8、int16、int32、int64、bool
torch.remainder	Stable	支持数据类型：fp16、fp32、int16、int32、int64
torch.round	Beta	不支持out；支持数据类型：fp16、fp32
torch.rsqrt	Stable	支持数据类型：bf16、fp16、fp32、uint8、int8、int16、int32、int64、bool
torch.sigmoid	Stable	支持数据类型：fp16、fp32、uint8、int8、int16、int32、int64、bool
torch.sign	Stable	支持数据类型：bf16、fp16、fp32、int32、int64、bool
torch.sgn	Not Support	N/A
torch.signbit	Not Support	N/A
torch.sin	Stable	支持数据类型：bf16、fp16、fp32、uint8、int8、int16、int32、int64、bool
torch.sinc	Stable	N/A
torch.sinh	Stable	支持数据类型：fp16、fp32、fp64
torch.softmax	Stable	支持数据类型：fp32
torch.sqrt	Stable	支持数据类型：bf16、fp16、fp32、fp64、uint8、int8、int16、int32、int64、bool
torch.square	Stable	支持数据类型：fp16、fp32、fp64、uint8、int8、int16、int32、int64
torch.sub	Stable	支持数据类型：bf16、fp16、fp32、fp64、uint8、int8、int16、int32、int64
torch.subtract	Beta	不支持out、alpha
torch.tan	Stable	支持数据类型：bf16、fp16、fp32、fp64、int64、uint8、int8、int16、int32、bool
torch.tanh	Stable	支持数据类型：bf16、fp16、fp32、uint8、int8、int16、int32、int64、bool
torch.true_divide	Beta	不支持out出参；支持数据类型：bf16、fp16、fp32、uint8、int8、int16、int32、int64、bool
torch.trunc	Stable	支持数据类型：fp16、fp32
torch.xlogy	Stable	支持数据类型：fp16、fp32、uint8、int8、int16、int32、int64、bool

Reduction Ops

API名称	API状态	限制与说明
torch.argmax	Stable	支持数据类型：bf16、fp16、fp32、fp64、uint8、int8、int16、int32、int64
torch.argmin	Stable	支持数据类型：fp16、fp32、uint8、int8、int16、int32、int64
torch.amax	Beta	不支持out出参；支持数据类型：fp16、fp32、uint8、int8、int16、int32、int64、bool
torch.amin	Beta	不支持out出参；支持数据类型：fp16、fp32、uint8、int8、int16、int32、int64、bool
torch.aminmax	Beta	不支持out出参；支持数据类型：fp16、fp32、uint8、int8、int16、int32、int64、bool
torch.all	Stable	支持数据类型：bf16、fp16、fp32、fp64、uint8、int8、int16、int32、int64、bool
torch.any	Stable	dim入参具有默认值None；支持数据类型：bf16、fp16、fp32、fp64、uint8、int8、int16、int32、int64、bool
torch.max	Stable	支持数据类型：bf16、fp16、fp32、int64、bool
torch.min	Stable	支持数据类型：bf16、fp16、fp32、int64、bool
torch.dist	Not Support	N/A
torch.logsumexp	Beta	不支持out出参；支持数据类型：fp16、fp32、uint8、int8、int16、int32、int64、bool
torch.mean	Stable	支持数据类型：bf16、fp16、fp32
torch.nanmean	Beta	不支持out出参
torch.median	Beta	不支持out出参；支持数据类型：bf16、fp16、fp32、uint8、int8、int16、int32、int64
torch.nanmedian	Not Support	N/A
torch.mode	Not Support	N/A
torch.norm	Stable	支持数据类型：支持bf16、fp16、fp32
torch.nansum	Beta	支持数据类型：bf16、fp16、fp32、uint8、int8、int16、int32、int64、bool
torch.prod	Stable	支持数据类型：fp16、fp32、uint8、int8、int16、int32、int64、bool
torch.quantile	Beta	不支持out出参
torch.nanquantile	Beta	不支持out出参
torch.std	Beta	不支持out出参；支持数据类型：fp16、fp32
torch.std_mean	Beta	不支持out出参
torch.sum	Beta	支持数据类型：bf16、fp16、fp32、fp64、uint8、int8、int16、int32、int64、bool
torch.unique	Stable	支持数据类型：fp16、fp32、uint8、int8、int16、int32、int64、bool
torch.unique_consecutive	Beta	N/A
torch.var	Beta	不支持out出参；支持数据类型：fp16、fp32
torch.var_mean	Beta	不支持out出参
torch.count_nonzero	Stable	支持数据类型：bf16、fp16、fp32、uint8、int8、int16、int32、int64、bool

Comparison Ops

API名称	API状态	限制与说明
torch.allclose	Beta	支持数据类型：fp16、fp32、uint8、int8、int16、int32、int64、bool
torch.argsort	Beta	支持数据类型：fp16、fp32、uint8、int8、int16、int32、int64
torch.eq	Stable	支持数据类型：bf16、fp16、fp32、uint8、int8、int16、int32、int64、bool
torch.equal	Stable	支持数据类型：fp16、fp32、uint8、int8、int16、int32、int64、bool
torch.ge	Beta	不支持out出参；支持数据类型：bf16、fp16、fp32、uint8、int8、int16、int32、int64、bool
torch.greater_equal	Not Support	N/A
torch.gt	Stable	支持数据类型：bf16、fp16、fp32、uint8、int8、int16、int32、int64、bool
torch.greater	Stable	支持数据类型：bf16、fp16、fp32、uint8、int8、int16、int32、int64、bool
torch.isclose	Stable	支持数据类型：fp16、fp32、uint8、int8、int16、int32、int64、bool
torch.isfinite	Stable	支持数据类型：bf16、fp16、fp32、fp64、uint8、int8、int16、int32、int64、bool
torch.isin	Beta	入参不支持assume_unique、invert
torch.isinf	Beta	支持数据类型：bf16、fp16、fp32、uint8、int8、int16、int32、int64、bool
torch.isposinf	Not Support	N/A
torch.isneginf	Not Support	N/A
torch.isnan	Beta	支持数据类型：bf16、fp16、fp32、uint8、int8、int16、int32、int64、bool
torch.isreal	Not Support	N/A
torch.kthvalue	Not Support	N/A
torch.le	Stable	支持数据类型：bf16、fp16、fp32、uint8、int8、int16、int32、int64、bool
torch.less_equal	Stable	支持数据类型：fp16、fp32、fp64、uint8、int8、int16、int32、int64
torch.lt	Stable	支持数据类型：bf16、fp16、fp32、uint8、int8、int16、int32、int64、bool
torch.less	Stable	支持数据类型：bf16、fp16、fp32、uint8、int8、int16、int32、int64、bool
torch.maximum	Stable	支持数据类型：bf16、fp16、fp32、uint8、int8、int16、int32、int64、bool
torch.minimum	Stable	支持数据类型：bf16、fp16、fp32、uint8、int8、int16、int32、int64、bool
torch.fmax	Beta	不支持out出参
torch.fmin	Beta	不支持out出参
torch.ne	Stable	支持数据类型：bf16、fp16、fp32、uint8、int8、int16、int32、int64、bool
torch.not_equal	Beta	不支持out出参；支持数据类型：bf16、fp16、fp32、uint8、int8、int16、int32、int64、bool
torch.sort	Beta	不支持out；支持数据类型：fp16、fp32、uint8、int8、int16、int32、int64
torch.topk	Beta	不支持out；支持数据类型：bf16、fp16、fp32、fp64、uint8、int8、int16、int32、int64
torch.msort	Beta	不支持out出参；支持数据类型：bf16、fp16、fp32、uint8、int8、int16、int32、int64

Spectral Ops

API名称	API状态	限制与说明
torch.stft	Beta	N/A
torch.istft	Not Support	N/A
torch.bartlett_window	Not Support	N/A
torch.blackman_window	Not Support	N/A
torch.hamming_window	Not Support	N/A
torch.hann_window	Beta	不支持out、layout、device、require_grad参数；支持数据类型：bf16、fp16、fp32
torch.kaiser_window	Not Support	N/A

Other Operations

API名称	API状态	限制与说明
torch.atleast_1d	Not Support	N/A
torch.atleast_2d	Not Support	N/A
torch.atleast_3d	Not Support	N/A
torch.bincount	Beta	不支持out出参；支持数据类型：uint8、int8、int16、int32、int64
torch.block_diag	Not Support	N/A
torch.broadcast_tensors	Beta	支持数据类型：bf16、fp16、fp32、uint8、int8、int16、int32、int64、bool
torch.broadcast_to	Stable	支持数据类型：bf16、fp16、fp32、fp64、uint8、int8、int16、int32、int64、bool
torch.broadcast_shapes	Beta	N/A
torch.bucketize	Not Support	N/A
torch.cartesian_prod	Demo	N/A
torch.cdist	Beta	N/A
torch.clone	Beta	入参不支持memory_format；支持数据类型：bf16、fp16、fp32、fp64、uint8、int8、int16、int32、int64、bool
torch.combinations	Not Support	N/A
torch.corrcoef	Not Support	N/A
torch.cov	Not Support	N/A
torch.cross	Not Support	N/A
torch.cummax	Not Support	N/A
torch.cummin	Not Support	N/A
torch.cumprod	Stable	不支持bfloat16输入
torch.cumsum	Stable	支持数据类型：fp16、fp32、uint8、int8、int16、int32、int64、bool
torch.diag	Beta	入参不支持diagnoal、out；支持数据类型：fp16、fp32、uint8、int8、int16、int32、int64、bool
torch.diag_embed	Not Support	N/A
torch.diagflat	Not Support	N/A
torch.diagonal	Not Support	N/A
torch.diff	Not Support	N/A
torch.einsum	Stable	支持数据类型：bf16、fp16、fp32、uint8、int8、int16、int32、int64、bool
torch.flatten	Stable	默认值start_dim为1，torch为0；支持数据类型：bf16、fp16、fp32、fp64、uint8、int8、int16、int32、int64、bool
torch.flip	Stable	支持数据类型：fp16、fp32、uint8、int8、int16、int32、int64、bool
torch.fliplr	Not Support	N/A
torch.flipud	Not Support	N/A
torch.kron	Not Support	N/A
torch.rot90	Not Support	N/A
torch.gcd	Not Support	N/A
torch.histc	Stable	支持数据类型：fp16、fp32
torch.histogram	Not Support	N/A
torch.histogramdd	Not Support	N/A
torch.meshgrid	Beta	支持数据类型：bf16、fp16、fp32、fp64、uint8、int8、int16、int32、int64、bool
torch.lcm	Beta	N/A
torch.logcumsumexp	Not Support	N/A
torch.ravel	Not Support	N/A
torch.renorm	Not Support	N/A
torch.repeat_interleave	Stable	支持数据类型：fp16、fp32、int16、int32、int64、bool
torch.roll	Stable	支持数据类型：fp16、fp32、int32、int64、bool
torch.searchsorted	Stable	支持数据类型：fp16、fp32、fp64、uint8、int8、int16、int32、int64
torch.tensordot	Not Support	N/A
torch.trace	Stable	N/A
torch.tril	Stable	支持数据类型：bf16、fp16、fp32、uint8、int8、int16、int32、int64、bool
torch.tril_indices	Not Support	N/A
torch.triu	Stable	支持数据类型：bf16、fp16、fp32、uint8、int8、int16、int32、int64、bool
torch.triu_indices	Not Support	N/A
torch.unflatten	Not Support	N/A
torch.vander	Not Support	N/A
torch.view_as_real	Not Support	N/A
torch.view_as_complex	Not Support	N/A
torch.resolve_conj	Not Support	N/A
torch.resolve_neg	Not Support	N/A

BLAS and LAPACK Operations

API名称	API状态	限制与说明
torch.addbmm	Stable	支持数据类型：fp16、fp32
torch.addmm	Beta	不支持out出参；支持数据类型：fp16、fp32
torch.addmv	Not Support	N/A
torch.addr	Not Support	N/A
torch.baddbmm	Stable	支持数据类型：fp16、fp32
torch.bmm	Stable	支持数据类型：fp16、fp32
torch.chain_matmul	Not Support	N/A
torch.cholesky	Not Support	N/A
torch.cholesky_inverse	Not Support	N/A
torch.cholesky_solve	Not Support	N/A
torch.dot	Beta	不支持out出参；支持数据类型：fp16、fp32
torch.geqrf	Not Support	N/A
torch.ger	Not Support	N/A
torch.inner	Not Support	N/A
torch.inverse	Stable	N/A
torch.det	Not Support	N/A
torch.logdet	Not Support	N/A
torch.slogdet	Not Support	N/A
torch.lu	Not Support	N/A
torch.lu_solve	Not Support	N/A
torch.lu_unpack	Not Support	N/A
torch.matmul	Stable	支持数据类型：fp16、fp32
torch.matrix_power	Not Support	N/A
torch.matrix_exp	Not Support	N/A
torch.mm	Beta	不支持out出参；支持数据类型：fp16、fp32
torch.mv	Not Support	N/A
torch.orgqr	Not Support	N/A
torch.ormqr	Not Support	N/A
torch.outer	Stable	支持数据类型：bf16、fp16、fp32、fp64、uint8、int8、int16、int32、int64、bool
torch.pinverse	Not Support	N/A
torch.qr	Not Support	N/A
torch.svd	Not Support	N/A
torch.svd_lowrank	Not Support	N/A
torch.pca_lowrank	Not Support	N/A
torch.lobpcg	Not Support	N/A
torch.trapz	Not Support	N/A
torch.trapezoid	Not Support	N/A
torch.cumulative_trapezoid	Not Support	N/A
torch.triangular_solve	Not Support	N/A
torch.vdot	Not Support	N/A

Foreach Operations

暂不支持

Utilities

API名称	API状态	限制与说明
torch.compiled_with_cxx11_abi	Not Support	N/A
torch.result_type	Not Support	N/A
torch.can_cast	Not Support	N/A
torch.promote_types	Not Support	N/A
torch.use_deterministic_algorithms	Not Support	N/A
torch.are_deterministic_algorithms_enabled	Not Support	N/A
torch.is_deterministic_algorithms_warn_only_enabled	Not Support	N/A
torch.set_deterministic_debug_mode	Not Support	N/A
torch.get_deterministic_debug_mode	Not Support	N/A
torch.set_float32_matmul_precision	Not Support	N/A
torch.get_float32_matmul_precision	Not Support	N/A
torch.set_warn_always	Not Support	N/A
torch.get_device_module	Not Support	N/A
torch.is_warn_always_enabled	Not Support	N/A
torch.vmap	Not Support	N/A
torch._assert	Not Support	N/A