torch
API名称  | 
是否支持  | 
限制与说明  | 
|---|---|---|
torch.SymInt  | 
是  | 
支持fp32  | 
torch.SymFloat  | 
是  | 
支持fp32  | 
torch.SymBool  | 
是  | 
支持fp32  | 
torch.Tag  | 
是  | 
|
torch.is_tensor  | 
是  | 
支持fp32  | 
torch.is_storage  | 
是  | 
|
torch.is_complex  | 
是  | 
支持fp32  | 
torch.is_conj  | 
否  | 
|
torch.is_floating_point  | 
是  | 
支持fp32  | 
torch.is_nonzero  | 
是  | 
支持fp32  | 
torch.set_default_dtype  | 
是  | 
|
torch.get_default_dtype  | 
是  | 
|
torch.set_default_device  | 
否  | 
|
torch.set_default_tensor_type  | 
是  | 
不支持传入torch.npu.DtypeTensor类型  | 
torch.numel  | 
是  | 
|
torch.set_printoptions  | 
是  | 
|
torch.set_flush_denormal  | 
是  | 
|
torch.tensor  | 
是  | 
支持fp32  | 
torch.asarray  | 
是  | 
支持bf16,fp16,fp32,fp64,uint8,int8,int16,int32,int64,bool  | 
torch.as_tensor  | 
是  | 
支持bf16,fp16,fp32,fp64,uint8,int8,int16,int32,int64,bool  | 
torch.as_strided  | 
是  | 
支持fp32  | 
torch.from_numpy  | 
是  | 
支持输出fp16,fp32,fp64,uint8,int8,int16,int32,int64,bool  | 
torch.frombuffer  | 
是  | 
支持bf16,fp16,fp32,fp64,uint8,int8,int16,int32,int64,bool  | 
torch.zeros  | 
是  | 
|
torch.zeros_like  | 
是  | 
支持bf16,fp16,fp32,uint8,int8,int16,int32,int64,bool,complex64,complex128  | 
torch.ones  | 
是  | 
支持bf16,fp16,fp32,fp64,uint8,int8,int16,int32,int64,bool  | 
torch.ones_like  | 
是  | 
支持bf16,fp16,fp32,fp64,uint8,int8,int16,int32,int64,bool  | 
torch.arange  | 
是  | 
支持bf16,fp16,fp32,fp64,int32,int64  | 
torch.range  | 
是  | 
|
torch.linspace  | 
是  | 
支持bf16,fp16,fp32,fp64  | 
torch.eye  | 
是  | 
支持fp16,fp32  | 
torch.empty  | 
是  | 
支持bf16,fp16,fp32,fp64,uint8,int8,int16,int32,int64,bool  | 
torch.empty_like  | 
是  | 
支持bf16,uint8,int8,int16,int32,int64,bool,complex64,complex128  | 
torch.empty_strided  | 
是  | 
|
torch.full  | 
是  | 
支持fp32  | 
torch.full_like  | 
是  | 
支持uint8,int8,int16,int32,int64,bool,complex64,complex128  | 
torch.quantize_per_tensor  | 
是  | 
|
torch.quantize_per_channel  | 
是  | 
|
torch.dequantize  | 
是  | 
支持fp16,fp32,fp64,uint8,int8,int16,int32,int64,bool  | 
torch.polar  | 
否  | 
|
torch.heaviside  | 
否  | 
|
torch.argwhere  | 
是  | 
支持fp16,fp32,uint8,int8,int16,int32,int64,bool  | 
torch.cat  | 
是  | 
支持bf16,fp16,fp32,fp64,uint8,int8,int16,int32,int64,bool  | 
torch.concat  | 
是  | 
支持bf16,fp16,fp32,uint8,int8,int16,int32,int64,bool,complex64  | 
torch.concatenate  | 
是  | 
支持bf16,fp16,fp32,int64,bool  | 
torch.conj  | 
是  | 
支持bf16,fp16,fp32,uint8,int8,int16,int32,int64,bool  | 
torch.chunk  | 
是  | 
支持bf16,fp16,fp32,fp64,uint8,int8,int16,int32,int64,bool,complex64,complex128  | 
torch.dsplit  | 
是  | 
支持bf16,fp16,fp32,uint8,int8,int16,int32,int64,bool,complex64,complex128  | 
torch.dstack  | 
是  | 
支持bf16,fp16,fp32,uint8,int8,int16,int32,int64,bool,complex64  | 
torch.gather  | 
是  | 
支持fp16,fp32,int16,int32,int64,bool  | 
torch.hsplit  | 
是  | 
支持bf16,fp16,fp32,uint8,int8,int16,int32,int64,bool,complex64,complex128  | 
torch.hstack  | 
是  | 
支持bf16,fp16,fp32,uint8,int8,int16,int32,int64,bool,complex64  | 
torch.index_add  | 
是  | 
支持fp16,fp32,int64,bool  | 
torch.index_copy  | 
是  | 
支持fp32  | 
torch.index_reduce  | 
是  | 
支持bf16,fp16,fp32,int64,bool  | 
torch.index_select  | 
是  | 
支持bf16,fp16,fp32,int16,int32,int64,bool  | 
torch.masked_select  | 
是  | 
支持fp16,fp32,int16,int32,int64,bool  | 
torch.movedim  | 
是  | 
支持bf16,fp16,fp32,fp64,uint8,int8,int16,int32,int64,bool  | 
torch.moveaxis  | 
是  | 
支持torch.int64, torch.float, torch.complex128  | 
torch.narrow  | 
是  | 
支持bf16,fp16,fp32,fp64,uint8,int8,int16,int32,int64,bool  | 
torch.narrow_copy  | 
是  | 
支持bf16,fp16,fp32,fp64,uint8,int8,int16,int32,int64,bool  | 
torch.nonzero  | 
是  | 
支持fp16,fp32,uint8,int8,int16,int32,int64,bool  | 
torch.permute  | 
是  | 
支持bf16,fp16,fp32,uint8,int8,int16,int32,int64,bool,complex64,complex128  | 
torch.reshape  | 
是  | 
支持bf16,fp16,fp32,fp64,uint8,int8,int16,int32,int64,bool,complex64,complex128  | 
torch.row_stack  | 
是  | 
支持bf16,fp16,fp32,uint8,int8,int16,int32,int64,bool,complex64  | 
torch.scatter  | 
是  | 
支持int16,int32,int64  | 
torch.diagonal_scatter  | 
是  | 
支持bf16,fp16,fp32,int16,int32,int64,bool  | 
torch.select_scatter  | 
是  | 
支持bf16,fp16,fp32,uint8,int8,int16,int32,int64,bool  | 
torch.slice_scatter  | 
否  | 
|
torch.split  | 
是  | 
支持bf16,fp16,fp32,uint8,int8,int16,int32,int64,bool,complex64,complex128  | 
torch.squeeze  | 
是  | 
支持bf16,fp16,fp32,fp64,uint8,int8,int16,int32,int64,bool  | 
torch.stack  | 
是  | 
支持bf16,fp16,fp32,fp64,uint8,int8,int16,int32,int64,bool  | 
torch.swapaxes  | 
是  | 
支持bf16,fp16,fp32,fp64,uint8,int8,int16,int32,int64,bool  | 
torch.swapdims  | 
是  | 
支持bf16,fp16,fp32,fp64,uint8,int8,int16,int32,int64,bool  | 
torch.t  | 
是  | 
支持bf16,fp16,fp32,fp64,uint8,int8,int16,int32,int64,bool  | 
torch.take  | 
是  | 
支持fp16,fp32,int16,int32,int64,bool  | 
torch.take_along_dim  | 
是  | 
支持fp32  | 
torch.tensor_split  | 
是  | 
支持bf16,fp16,fp32,uint8,int8,int16,int32,int64,bool,complex64,complex128  | 
torch.tile  | 
是  | 
支持bf16,fp16,fp32,fp64,uint8,int8,int16,int32,int64,bool  | 
torch.transpose  | 
是  | 
支持bf16,fp16,fp32,uint8,int8,int16,int32,int64,bool,complex64,complex128  | 
torch.unsqueeze  | 
是  | 
支持bf16,fp16,fp32,fp64,uint8,int8,int16,int32,int64,bool  | 
torch.vsplit  | 
是  | 
支持bf16,fp16,fp32,uint8,int8,int16,int32,int64,bool,complex64,complex128  | 
torch.vstack  | 
是  | 
支持bf16,fp16,fp32,fp64,uint8,int8,int16,int32,int64,bool  | 
torch.where  | 
是  | 
支持bf16,fp16,fp32,fp64,uint8,int8,int16,int32,int64,bool 不支持8维度的shape  | 
torch.Generator  | 
是  | 
|
torch.Generator.get_state  | 
是  | 
|
torch.Generator.initial_seed  | 
是  | 
|
torch.Generator.manual_seed  | 
是  | 
|
torch.Generator.seed  | 
是  | 
|
torch.Generator.set_state  | 
是  | 
|
torch.seed  | 
是  | 
|
torch.manual_seed  | 
是  | 
|
torch.initial_seed  | 
是  | 
|
torch.get_rng_state  | 
是  | 
|
torch.set_rng_state  | 
是  | 
|
torch.bernoulli  | 
是  | 
支持fp32  | 
torch.multinomial  | 
是  | 
支持fp16,fp32  | 
torch.normal  | 
是  | 
支持fp16,fp32  | 
torch.poisson  | 
否  | 
|
torch.rand  | 
是  | 
|
torch.rand_like  | 
是  | 
支持uint8,int8,int16,int32,int64  | 
torch.randint  | 
是  | 
|
torch.randint_like  | 
是  | 
支持fp16,fp32,int64  | 
torch.randn  | 
是  | 
|
torch.randn_like  | 
是  | 
支持fp32  | 
torch.randperm  | 
是  | 
|
torch.save  | 
是  | 
|
torch.load  | 
是  | 
|
torch.get_num_threads  | 
是  | 
|
torch.set_num_threads  | 
是  | 
|
torch.get_num_interop_threads  | 
是  | 
|
torch.set_num_interop_threads  | 
是  | 
|
torch.no_grad  | 
是  | 
|
torch.enable_grad  | 
是  | 
|
torch.autograd.grad_mode.set_grad_enabled  | 
是  | 
|
torch.is_grad_enabled  | 
是  | 
|
torch.autograd.grad_mode.inference_mode  | 
是  | 
|
torch.is_inference_mode_enabled  | 
是  | 
|
torch.abs  | 
是  | 
支持bf16,fp16,fp32,fp64,uint8,int8,int16,int32,int64,bool  | 
torch.absolute  | 
是  | 
支持bf16,fp16,fp32,uint8,int8,int16,int32,int64  | 
torch.acos  | 
是  | 
支持fp16,fp32,uint8,int8,int16,int32,int64,bool  | 
torch.arccos  | 
是  | 
支持fp16,fp32,uint8,int8,int16,int32,int64,bool  | 
torch.acosh  | 
是  | 
支持fp16,fp32,fp64,uint8,int8,int16,int32,int64,bool  | 
torch.arccosh  | 
是  | 
支持fp16,fp32,fp64,uint8,int8,int16,int32,int64,bool  | 
torch.add  | 
是  | 
支持bf16,fp16,fp32,fp64,uint8,int8,int16,int32,int64,bool  | 
torch.addcdiv  | 
是  | 
支持fp16,fp32,int64 在int64类型不支持三个tensor同时广播  | 
torch.addcmul  | 
是  | 
支持bf16,fp16,fp32,fp64,uint8,int8,int32,int64 在fp64,uint8,int8,int64类型不支持三个tensor同时广播  | 
torch.angle  | 
是  | 
支持fp16,fp32,fp64,uint8,int8,int16,int32,int64,bool  | 
torch.asin  | 
是  | 
支持bf16,fp16,fp32,uint8,int8,int16,int32,int64,bool  | 
torch.arcsin  | 
是  | 
支持bf16,fp16,fp32,uint8,int8,int16,int32,int64,bool  | 
torch.asinh  | 
是  | 
支持fp16,fp32,uint8,int8,int16,int32,int64,bool,complex64,complex128  | 
torch.arcsinh  | 
是  | 
支持fp16,fp32,uint8,int8,int16,int32,int64,bool,complex64,complex128  | 
torch.atan  | 
是  | 
支持fp16,fp32,uint8,int8,int16,int32,int64,bool  | 
torch.arctan  | 
是  | 
支持fp16,fp32,uint8,int8,int16,int32,int64,bool  | 
torch.atanh  | 
是  | 
支持fp16,fp32,uint8,int8,int16,int32,int64,bool,complex64,complex128  | 
torch.arctanh  | 
是  | 
支持fp16,fp32,uint8,int8,int16,int32,int64,bool,complex64,complex128  | 
torch.atan2  | 
是  | 
支持fp16,fp32,uint8,int8,int16,int32,int64,bool  | 
torch.arctan2  | 
是  | 
支持fp16,fp32,uint8,int8,int16,int32,int64,bool  | 
torch.bitwise_not  | 
是  | 
支持uint8,int8,int16,int32,int64,bool  | 
torch.bitwise_and  | 
是  | 
支持uint8,int8,int16,int32,int64,bool  | 
torch.bitwise_or  | 
是  | 
支持uint8,int8,int16,int32,int64,bool  | 
torch.bitwise_xor  | 
是  | 
支持uint8,int8,int16,int32,int64,bool  | 
torch.ceil  | 
是  | 
支持fp16,fp32  | 
torch.clamp  | 
是  | 
支持bf16,fp16,fp32,uint8,int8,int16,int32,int64  | 
torch.clip  | 
是  | 
支持bf16,fp16,fp32,uint8,int8,int16,int32,int64  | 
torch.copysign  | 
是  | 
支持bf16,fp16,fp32,uint8,int8,int16,int32,int64,bool  | 
torch.cos  | 
是  | 
支持bf16,fp16,fp32,uint8,int8,int16,int32,int64,bool,complex64,complex128  | 
torch.cosh  | 
是  | 
支持fp16,fp32,fp64  | 
torch.deg2rad  | 
是  | 
支持fp16,fp32,fp64,uint8,int8,int16,int32,int64,bool  | 
torch.div  | 
是  | 
支持bf16,fp16,fp32,uint8,int8,int16,int32,int64,bool  | 
torch.divide  | 
是  | 
支持bf16,fp16,fp32,uint8,int8,int16,int32,int64,bool  | 
torch.erf  | 
是  | 
支持fp16,fp32,int64,bool  | 
torch.erfc  | 
是  | 
支持fp16,fp32,int64,bool  | 
torch.erfinv  | 
是  | 
支持fp16,fp32,uint8,int8,int16,int32,int64,bool  | 
torch.exp  | 
是  | 
支持bf16,fp16,fp32,fp64,int64,bool,complex64,complex128  | 
torch.exp2  | 
是  | 
支持bf16,fp16,fp32,uint8,int8,int16,int32,int64,bool  | 
torch.expm1  | 
是  | 
支持fp16,fp32,int64,bool  | 
torch.fix  | 
是  | 
支持fp16,fp32  | 
torch.float_power  | 
是  | 
支持bf16,fp16,fp32,uint8,int8,int16,int32,int64,bool,complex128  | 
torch.floor  | 
是  | 
支持fp16,fp32  | 
torch.floor_divide  | 
是  | 
支持fp16,fp32,uint8,int8,int16,int32,int64  | 
torch.fmod  | 
是  | 
支持fp16,fp32,uint8,int8,int32,int64  | 
torch.gradient  | 
是  | 
支持bf16,fp16,fp32,int8,int16,int32,int64  | 
torch.ldexp  | 
是  | 
支持fp16,fp64,complex64  | 
torch.lerp  | 
是  | 
支持fp16,fp32  | 
torch.log  | 
是  | 
支持bf16,fp16,fp32,fp64,uint8,int8,int16,int32,int64,bool,complex64,complex128  | 
torch.log10  | 
是  | 
支持fp16,fp32,fp64,uint8,int8,int16,int32,int64,bool  | 
torch.log1p  | 
是  | 
支持fp16,fp32,uint8,int8,int16,int32,int64,bool  | 
torch.log2  | 
是  | 
支持bf16,fp32,int64,bool,fp16  | 
torch.logaddexp  | 
是  | 
不支持double数据类型  | 
torch.logaddexp2  | 
是  | 
不支持double数据类型  | 
torch.logical_and  | 
是  | 
支持bf16,fp16,fp32,uint8,int8,int16,int32,int64,bool,complex64,complex128  | 
torch.logical_not  | 
是  | 
支持fp16,fp32,uint8,int8,int16,int32,int64,bool  | 
torch.logical_or  | 
是  | 
支持bf16,fp16,fp32,uint8,int8,int16,int32,int64,bool,complex64,complex128  | 
torch.logical_xor  | 
是  | 
支持fp16,fp32,uint8,int8,int16,int32,int64,bool,complex64,complex128  | 
torch.mul  | 
是  | 
支持bf16,fp16,fp32,uint8,int8,int16,int32,int64,bool,complex64,complex128  | 
torch.multiply  | 
是  | 
支持bf16,fp16,fp32,uint8,int8,int16,int32,int64,bool,complex64,complex128  | 
torch.nan_to_num  | 
是  | 
支持bf16,fp16,fp32,uint8,int8,int16,int32,int64,bool  | 
torch.neg  | 
是  | 
支持bf16,fp16,fp32,int8,int32,int64,complex64,complex128  | 
torch.negative  | 
是  | 
支持bf16,fp16,fp32,int8,int32,int64,complex64,complex128  | 
torch.positive  | 
是  | 
支持bf16,fp16,fp32,uint8,int8,int16,int32,int64,complex64,complex128  | 
torch.pow  | 
是  | 
支持bf16,fp16,fp32,fp64,int16,int64  | 
torch.rad2deg  | 
是  | 
支持fp16,fp32,fp64,uint8,int8,int16,int32,int64,bool  | 
torch.real  | 
是  | 
支持fp16,fp32,complex64,complex128  | 
torch.reciprocal  | 
是  | 
支持bf16,fp16,fp32,uint8,int8,int16,int32,int64,bool,complex64,complex128  | 
torch.remainder  | 
是  | 
支持fp16,fp32,int16,int32,int64  | 
torch.round  | 
是  | 
支持fp16,fp32  | 
torch.rsqrt  | 
是  | 
支持bf16,fp16,fp32,uint8,int8,int16,int32,int64,bool,complex64,complex128  | 
torch.sigmoid  | 
是  | 
支持fp16,fp32,uint8,int8,int16,int32,int64,bool,complex64,complex128  | 
torch.sign  | 
是  | 
支持bf16,fp16,fp32,int32,int64,bool  | 
torch.sgn  | 
是  | 
支持bf16,fp16,fp32,int32,int64,bool,complex64,complex128  | 
torch.sin  | 
是  | 
支持bf16,fp16,fp32,uint8,int8,int16,int32,int64,bool,complex64,complex128  | 
torch.sinh  | 
是  | 
支持fp16,fp32,fp64  | 
torch.softmax  | 
是  | 
支持fp32 支持Named Tensor  | 
torch.sqrt  | 
是  | 
支持bf16,fp16,fp32,fp64,uint8,int8,int16,int32,int64,bool  | 
torch.square  | 
是  | 
支持fp16,fp32,fp64,uint8,int8,int16,int32,int64  | 
torch.sub  | 
是  | 
支持bf16,fp16,fp32,fp64,uint8,int8,int16,int32,int64  | 
torch.tan  | 
是  | 
支持fp16,fp32,fp64  | 
torch.tanh  | 
是  | 
支持bf16,fp16,fp32,uint8,int8,int16,int32,int64,bool  | 
torch.true_divide  | 
是  | 
支持bf16,fp16,fp32,uint8,int8,int16,int32,int64,bool  | 
torch.trunc  | 
是  | 
支持fp16,fp32  | 
torch.xlogy  | 
是  | 
支持fp16,fp32,uint8,int8,int16,int32,int64,bool  | 
torch.argmax  | 
是  | 
支持bf16,fp16,fp32,fp64,uint8,int8,int16,int32,int64  | 
torch.argmin  | 
是  | 
支持fp16,fp32,uint8,int8,int16,int32,int64  | 
torch.amax  | 
是  | 
支持fp16,fp32,uint8,int8,int16,int32,int64,bool  | 
torch.amin  | 
是  | 
支持fp16,fp32,uint8,int8,int16,int32,int64,bool  | 
torch.aminmax  | 
是  | 
支持fp16,fp32,uint8,int8,int16,int32,int64,bool  | 
torch.all  | 
是  | 
支持bf16,fp16,fp32,fp64,uint8,int8,int16,int32,int64,bool  | 
torch.any  | 
是  | 
支持bf16,fp16,fp32,fp64,uint8,int8,int16,int32,int64,bool  | 
torch.max  | 
是  | 
支持bf16,fp16,fp32,int64,bool  | 
torch.min  | 
是  | 
支持bf16,fp16,fp32,int64,bool  | 
torch.dist  | 
是  | 
支持bf16,fp16,fp32  | 
torch.logsumexp  | 
是  | 
支持fp16,fp32,uint8,int8,int16,int32,int64,bool  | 
torch.mean  | 
是  | 
支持bf16,fp16,fp32,complex64,complex128  | 
torch.median  | 
是  | 
支持bf16,fp16,fp32,uint8,int8,int16,int32,int64  | 
torch.norm  | 
是  | 
支持bf16,fp16,fp32,complex128 参数p为负数时,计算结果可能存在精度误差 参数dim指定为输入tensor中shape维度值为1的轴时,计算结果可能存在精度误差  | 
torch.nansum  | 
是  | 
支持bf16,fp16,fp32,uint8,int8,int16,int32,int64,bool  | 
torch.prod  | 
是  | 
支持fp16,fp32,uint8,int8,int16,int32,int64,bool,complex64,complex128  | 
torch.nanquantile  | 
否  | 
|
torch.std  | 
是  | 
支持fp16,fp32  | 
torch.std_mean  | 
否  | 
|
torch.sum  | 
是  | 
支持bf16,fp16,fp32,fp64,uint8,int8,int16,int32,int64,bool 不支持dtype参数  | 
torch.unique  | 
是  | 
支持fp16,fp32,uint8,int8,int16,int32,int64,bool 带dim场景不支持fp16  | 
torch.unique_consecutive  | 
否  | 
|
torch.var  | 
是  | 
支持fp16,fp32  | 
torch.var_mean  | 
否  | 
|
torch.count_nonzero  | 
是  | 
支持bf16,fp16,fp32,uint8,int8,int16,int32,int64,bool,complex64,complex128  | 
torch.allclose  | 
是  | 
支持fp16,fp32,uint8,int8,int16,int32,int64,bool  | 
torch.argsort  | 
是  | 
支持fp16,fp32,uint8,int8,int16,int32,int64  | 
torch.eq  | 
是  | 
支持bf16,fp16,fp32,uint8,int8,int16,int32,int64,bool,complex64,complex128  | 
torch.equal  | 
是  | 
支持fp16,fp32,uint8,int8,int16,int32,int64,bool  | 
torch.ge  | 
是  | 
支持bf16,fp16,fp32,uint8,int8,int16,int32,int64,bool  | 
torch.greater_equal  | 
是  | 
支持bf16,fp16,fp32,uint8,int8,int16,int32,int64,bool  | 
torch.gt  | 
是  | 
支持bf16,fp16,fp32,uint8,int8,int16,int32,int64,bool  | 
torch.greater  | 
是  | 
支持bf16,fp16,fp32,uint8,int8,int16,int32,int64,bool  | 
torch.isclose  | 
是  | 
支持fp16,fp32,uint8,int8,int16,int32,int64,bool  | 
torch.isfinite  | 
是  | 
支持bf16,fp16,fp32,fp64,uint8,int8,int16,int32,int64,bool  | 
torch.isinf  | 
是  | 
支持bf16,fp16,fp32,uint8,int8,int16,int32,int64,bool,complex64,complex128  | 
torch.isposinf  | 
是  | 
支持bf16,fp16,fp32,uint8,int8,int16,int32,int64,bool  | 
torch.isneginf  | 
是  | 
支持bf16,fp16,fp32,uint8,int8,int16,int32,int64,bool  | 
torch.isnan  | 
是  | 
支持bf16,fp16,fp32,uint8,int8,int16,int32,int64,bool,complex64,complex128  | 
torch.kthvalue  | 
是  | 
支持fp16,fp32和int32  | 
torch.le  | 
是  | 
支持bf16,fp16,fp32,uint8,int8,int16,int32,int64,bool  | 
torch.less_equal  | 
是  | 
支持fp16,fp32,fp64,uint8,int8,int16,int32,int64  | 
torch.lt  | 
是  | 
支持bf16,fp16,fp32,uint8,int8,int16,int32,int64,bool  | 
torch.less  | 
是  | 
支持bf16,fp16,fp32,uint8,int8,int16,int32,int64,bool  | 
torch.maximum  | 
是  | 
支持bf16,fp16,fp32,uint8,int8,int16,int32,int64,bool  | 
torch.minimum  | 
是  | 
支持bf16,fp16,fp32,uint8,int8,int16,int32,int64,bool  | 
torch.ne  | 
是  | 
支持bf16,fp16,fp32,uint8,int8,int16,int32,int64,bool,complex64,complex128  | 
torch.not_equal  | 
是  | 
支持bf16,fp16,fp32,uint8,int8,int16,int32,int64,bool,complex64,complex128  | 
torch.sort  | 
是  | 
支持fp16,fp32,uint8,int8,int16,int32,int64  | 
torch.topk  | 
是  | 
支持bf16,fp16,fp32,fp64,uint8,int8,int16,int32,int64 不支持sorted=False场景  | 
torch.msort  | 
是  | 
支持bf16,fp16,fp32,uint8,int8,int16,int32,int64  | 
torch.stft  | 
是  | 
若算子超时,需要用官方接口set_op_execute_time_out进行设置,调高超时阈值判断时间  | 
torch.hann_window  | 
是  | 
支持bf16,fp16,fp32  | 
torch.atleast_1d  | 
是  | 
支持bf16,fp16,fp32,fp64,uint8,int8,int16,int32,int64,bool  | 
torch.atleast_2d  | 
是  | 
支持fp16,fp32,fp64,uint8,int8,int16,int32,int64,bool  | 
torch.atleast_3d  | 
是  | 
支持fp16,fp32,fp64,uint8,int8,int16,int32,int64,bool  | 
torch.bincount  | 
是  | 
支持uint8,int8,int16,int32,int64  | 
torch.block_diag  | 
是  | 
支持bf16,fp16,fp32,uint8,int8,int16,int32,int64,bool,complex64,complex128  | 
torch.broadcast_tensors  | 
是  | 
支持bf16,fp16,fp32,uint8,int8,int16,int32,int64,bool,complex64,complex128  | 
torch.broadcast_to  | 
是  | 
支持bf16,fp16,fp32,fp64,uint8,int8,int16,int32,int64,bool  | 
torch.broadcast_shapes  | 
是  | 
|
torch.cdist  | 
否  | 
|
torch.clone  | 
是  | 
支持bf16,fp16,fp32,fp64,uint8,int8,int16,int32,int64,bool  | 
torch.combinations  | 
是  | 
支持fp16,fp32,fp64,uint8,int8,int16,int32,int64,bool  | 
torch.cov  | 
是  | 
支持fp32  | 
torch.cross  | 
是  | 
支持fp16,fp32,uint8,int8,int16,int32,int64,complex64,complex128 两个输入的shape要保持一致  | 
torch.cummax  | 
是  | 
支持fp16,fp32,uint8,int8,int16,int32,int64,bool  | 
torch.cummin  | 
是  | 
支持fp16,fp32,uint8,int8,int16,int32,int64,bool 输入为int32时,数值范围在[-16777216, 16777216]内  | 
torch.cumprod  | 
是  | 
支持fp16,fp32,int32  | 
torch.cumsum  | 
是  | 
支持fp16,fp32,uint8,int8,int16,int32,int64,bool,complex64,complex128 支持Named Tensor  | 
torch.diag  | 
是  | 
支持fp16,fp32,uint8,int8,int16,int32,int64,bool,complex64  | 
torch.diag_embed  | 
是  | 
支持bf16,fp16,fp32,uint8,int8,int16,int32,int64  | 
torch.diagonal  | 
是  | 
支持bf16,fp16,fp32,uint8,int8,int16,int32,int64,bool,complex64,complex128  | 
torch.diff  | 
是  | 
支持fp16,fp32,fp64,uint8,int8,int16,int32,int64,bool  | 
torch.einsum  | 
是  | 
支持bf16,fp16,fp32,uint8,int8,int16,int32,int64,bool  | 
torch.flatten  | 
是  | 
支持bf16,fp16,fp32,fp64,uint8,int8,int16,int32,int64,bool,complex64,complex128  | 
torch.flip  | 
是  | 
支持fp16,fp32,uint8,int8,int16,int32,int64,bool,complex64,complex128  | 
torch.fliplr  | 
是  | 
支持fp16,fp32,uint8,int8,int16,int32,int64,bool,complex64,complex128  | 
torch.flipud  | 
是  | 
支持fp16,fp32,uint8,int8,int16,int32,int64,bool,complex64,complex128  | 
torch.kron  | 
是  | 
不支持5维度及以上输入。  | 
torch.histc  | 
是  | 
支持fp16,fp32 当输入tensor值处于计数区间交界时,归于左区间计数还是右区间计数可能存在误差  | 
torch.meshgrid  | 
是  | 
支持bf16,fp16,fp32,fp64,uint8,int8,int16,int32,int64,bool  | 
torch.ravel  | 
是  | 
支持bf16,fp16,fp32,uint8,int8,int16,int32,int64,bool,complex64,complex128  | 
torch.repeat_interleave  | 
是  | 
支持fp16,fp32,int16,int32,int64,bool 输入张量在重复后得到输出,输出中元素个数需小于222  | 
torch.roll  | 
是  | 
支持fp16,fp32,int32,int64,bool  | 
torch.searchsorted  | 
是  | 
支持fp16,fp32,fp64,uint8,int8,int16,int32,int64  | 
torch.tensordot  | 
是  | 
支持fp16,fp32  | 
torch.tril  | 
是  | 
支持bf16,fp16,fp32,uint8,int8,int16,int32,int64,bool  | 
torch.tril_indices  | 
是  | 
|
torch.triu  | 
是  | 
支持bf16,fp16,fp32,uint8,int8,int16,int32,int64,bool  | 
torch.triu_indices  | 
是  | 
|
torch.unflatten  | 
是  | 
|
torch.view_as_real  | 
是  | 
支持complex64  | 
torch.resolve_conj  | 
是  | 
支持fp16,fp32,fp64,uint8,int8,int16,int32,int64,bool  | 
torch.resolve_neg  | 
是  | 
支持bf16,fp16,fp32,uint8,int8,int16,int32,int64,bool,complex64,complex128  | 
torch.addbmm  | 
是  | 
支持fp16,fp32  | 
torch.addmm  | 
是  | 
支持fp16,fp32  | 
torch.addmv  | 
是  | 
支持fp16,fp32,uint8,int8,int16,int32,int64  | 
torch.addr  | 
是  | 
支持fp16,fp32,uint8,int8,int16,int32,int64,bool  | 
torch.baddbmm  | 
是  | 
支持fp16,fp32  | 
torch.bmm  | 
是  | 
支持fp16,fp32  | 
torch.dot  | 
是  | 
支持fp16,fp32  | 
torch.slogdet  | 
是  | 
支持fp32,complex64,complex128  | 
torch.matmul  | 
是  | 
支持fp16,fp32 支持Named Tensor  | 
torch.mm  | 
是  | 
支持fp16,fp32  | 
torch.outer  | 
是  | 
支持bf16,fp16,fp32,fp64,uint8,int8,int16,int32,int64,bool  | 
torch.qr  | 
是  | 
|
torch.trapezoid  | 
是  | 
支持bf16,fp16,fp32,uint8,int8,int16,int32,int64  | 
torch.cumulative_trapezoid  | 
是  | 
支持fp16,fp32,fp64,uint8,int8,int16,int32,int64  | 
torch.vdot  | 
是  | 
支持fp16,fp32  | 
torch.compiled_with_cxx11_abi  | 
是  | 
|
torch.result_type  | 
是  | 
支持fp32  | 
torch.can_cast  | 
是  | 
|
torch.promote_types  | 
是  | 
|
torch.use_deterministic_algorithms  | 
是  | 
|
torch.are_deterministic_algorithms_enabled  | 
是  | 
|
torch.set_deterministic_debug_mode  | 
是  | 
|
torch.get_deterministic_debug_mode  | 
是  | 
|
torch.set_warn_always  | 
是  | 
|
torch.is_warn_always_enabled  | 
是  | 
|
torch.vmap  | 
是  | 
|
torch._assert  | 
是  | 
|
torch.sym_float  | 
是  | 
支持fp32  | 
torch.sym_int  | 
是  | 
支持fp32  | 
torch.sym_max  | 
是  | 
支持fp32  | 
torch.sym_min  | 
是  | 
支持fp32  | 
torch.sym_not  | 
是  | 
支持fp32  | 
torch.compile  | 
否  |