torch

torch.default_generator

torch.SymInt

是

支持fp32

torch.SymFloat

是

支持fp32

torch.SymBool

是

支持fp32

torch.Tag

是

torch.Tag.name

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.sparse_coo_tensor

torch.sparse_csr_tensor

torch.sparse_csc_tensor

torch.sparse_bsr_tensor

torch.sparse_bsc_tensor

torch.asarray

是

支持bf16，fp16，fp32，fp64，int8，uint8，int16，int32，int64，bool

torch.as_tensor

是

支持bf16，fp16，fp32，fp64，int8，uint8，int16，int32，int64，bool

torch.as_strided

是

支持fp32

torch.from_file

torch.from_numpy

是

支持输出fp16，fp32，fp64，uint8，int8，int16，int32，int64，bool

torch.from_dlpack

torch.frombuffer

是

支持bf16，fp16，fp32，fp64，int8，uint8，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，uint8，int8，int16，int32，int64

torch.range

是

torch.linspace

是

支持bf16，fp16，fp32，fp64

torch.logspace

torch.eye

是

支持fp16，fp32

torch.empty

是

支持bf16，fp16，fp32，fp64，int8，uint8，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，int8，uint8，int16，int32，int64，bool

torch.complex

torch.polar

否

torch.heaviside

否

torch.adjoint

torch.argwhere

是

支持fp32，fp16，uint8，int8，int16，int32，int64，bool

torch.cat

是

支持bf16，fp16，fp32，fp64，uint8，int8，int16，int32，int64，bool

torch.concat

是

支持bf16，fp32，fp16，uint8，int8，int16，int32，int64，bool，complex64

torch.concatenate

是

支持bf16，fp16，fp32，int64，bool

torch.conj

是

支持bf16，fp32，fp16，uint8，int8，int16，int32，int64，bool

torch.chunk

是

支持bf16，fp16，fp32，fp64，uint8，int8，int16，int32，int64，bool，complex64，complex128

torch.dsplit

是

支持bf16，fp32，fp16，uint8，int8，int16，int32，int64，bool，complex64，complex128

torch.column_stack

torch.dstack

是

支持bf16，fp32，fp16，uint8，int8，int16，int32，int64，bool，complex64

torch.gather

是

支持fp32，fp16，int16，int32，int64，bool

torch.hsplit

是

支持bf16，fp32，fp16，uint8，int8，int16，int32，int64，bool，complex64，complex128

torch.hstack

是

支持bf16，fp32，fp16，uint8，int8，int16，int32，int64，bool，complex64

torch.index_add

是

支持fp32，fp16，int64，bool

torch.index_copy

是

支持fp32

torch.index_reduce

是

支持bf16，fp16，fp32，int64，bool

torch.index_select

是

支持bf16，fp32，fp16，int16，int32，int64，bool

torch.masked_select

是

支持fp32，fp16，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，int8，uint8，int16，int32，int64，bool

torch.narrow_copy

是

支持bf16，fp16，fp32，fp64，uint8，int8，int16，int32，int64，bool

torch.nonzero

是

支持fp32，fp16，uint8，int8，int16，int32，int64，bool

torch.permute

是

支持bf16，fp32，fp16，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，fp32，fp16，uint8，int8，int16，int32，int64，bool，complex64

torch.select

torch.scatter

是

支持int16，int32，int64

torch.diagonal_scatter

是

支持bf16，fp32，fp16，int16，int32，int64，bool

torch.select_scatter

是

支持bf16，fp32，fp16，uint8，int8，int16，int32，int64，bool

torch.slice_scatter

否

torch.scatter_add

torch.scatter_reduce

torch.split

是

支持bf16，fp32，fp16，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，int8，uint8，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，int8，uint8，int16，int32，int64，bool

torch.take

是

支持fp32，fp16，int16，int32，int64，bool

torch.take_along_dim

是

支持fp32

torch.tensor_split

是

支持bf16，fp32，fp16，uint8，int8，int16，int32，int64，bool，complex64，complex128

torch.tile

是

支持bf16，fp16，fp32，fp64，uint8，int8，int16，int32，int64，bool

torch.transpose

是

支持bf16，fp32，fp16，uint8，int8，int16，int32，int64，bool，complex64，complex128

torch.unbind

torch.unravel_index

torch.unsqueeze

是

支持bf16，fp16，fp32，fp64，int8，uint8，int16，int32，int64，bool

torch.vsplit

是

支持bf16，fp32，fp16，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.device

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

是

支持fp32，fp16

torch.normal

是

支持fp32，fp16

torch.poisson

否

torch.rand

是

torch.rand_like

是

支持int8，uint8，int16，int32，int64

torch.randint

是

torch.randint_like

是

支持fp32，fp16，int64

torch.randn

是

torch.randn_like

是

支持fp32

torch.randperm

是

torch.quasirandom.SobolEngine

torch.quasirandom.SobolEngine.draw

torch.quasirandom.SobolEngine.draw_base2

torch.quasirandom.SobolEngine.fast_forward

torch.quasirandom.SobolEngine.reset

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.set_grad_enabled

是

torch.is_grad_enabled

是

torch.inference_mode

是

torch.is_inference_mode_enabled

是

torch.abs

是

支持bf16，fp32，fp16，fp64，uint8，int8，int16，int32，int64，bool

torch.absolute

是

支持bf16，fp32，fp16，uint8，int8，int16，int32，int64

torch.acos

是

支持fp32，fp16，uint8，int8，int16，int32，int64，bool

torch.arccos

是

支持fp32，fp16，uint8，int8，int16，int32，int64，bool

torch.acosh

是

支持fp16，fp32，fp64，int8，uint8，int16，int32，int64，bool

torch.arccosh

是

支持fp16，fp32，fp64，int8，uint8，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

在int8，uint8，int64，fp64类型不支持三个tensor同时广播

torch.angle

是

支持fp16，fp32，fp64，uint8，int8，int16，int32，int64，bool

torch.asin

是

支持bf16，fp32，fp16，uint8，int8，int16，int32，int64，bool

torch.arcsin

是

支持bf16，fp32，fp16，uint8，int8，int16，int32，int64，bool

torch.asinh

是

支持fp32，fp16，uint8，int8，int16，int32，int64，bool，complex64，complex128

torch.arcsinh

是

支持fp32，fp16，uint8，int8，int16，int32，int64，bool，complex64，complex128

torch.atan

是

支持fp32，fp16，uint8，int8，int16，int32，int64，bool

torch.arctan

是

支持fp32，fp16，uint8，int8，int16，int32，int64，bool

torch.atanh

是

支持fp32，fp16，uint8，int8，int16，int32，int64，bool，complex64，complex128

torch.arctanh

是

支持fp32，fp16，uint8，int8，int16，int32，int64，bool，complex64，complex128

torch.atan2

是

支持fp32，fp16，uint8，int8，int16，int32，int64，bool

torch.arctan2

是

支持fp32，fp16，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.bitwise_left_shift

torch.bitwise_right_shift

torch.ceil

是

支持fp32，fp16

torch.clamp

是

支持bf16，fp32，fp16，uint8，int8，int16，int32，int64

torch.clip

是

支持bf16，fp32，fp16，uint8，int8，int16，int32，int64

torch.conj_physical

torch.copysign

是

支持bf16，fp32，fp16，uint8，int8，int16，int32，int64，bool

torch.cos

是

支持bf16，fp32，fp16，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，fp32，fp16，uint8，int8，int16，int32，int64，bool

torch.divide

是

支持bf16，fp32，fp16，uint8，int8，int16，int32，int64，bool

torch.digamma

torch.erf

是

支持fp32，fp16，int64，bool

torch.erfc

是

支持fp32，fp16，int64，bool

torch.erfinv

是

支持fp32，fp16，uint8，int8，int16，int32，int64，bool

torch.exp

是

支持bf16，fp16，fp32，fp64，int64，bool，complex64，complex128

torch.exp2

是

支持bf16，fp32，fp16，uint8，int8，int16，int32，int64，bool

torch.expm1

是

支持fp32，fp16，int64，bool

torch.fake_quantize_per_channel_affine

torch.fake_quantize_per_tensor_affine

torch.fix

是

支持fp16，fp32

torch.float_power

是

支持bf16，fp32，fp16，uint8，int8，int16，int32，int64，bool，complex128

torch.floor

是

支持fp32，fp16

torch.floor_divide

是

支持fp32，fp16，uint8，int8，int16，int32，int64

torch.fmod

是

支持fp32，fp16，uint8，int8，int32，int64

torch.frac

torch.frexp

torch.gradient

是

支持bf16，fp32，fp16，int8，int16，int32，int64

torch.imag

torch.ldexp

是

支持fp16，fp64，complex64

torch.lerp

是

支持fp32，fp16

torch.lgamma

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

是

支持fp32，fp16，uint8，int8，int16，int32，int64，bool

torch.log2

是

支持bf16，fp32，int64，bool，fp16

torch.logaddexp

是

不支持double数据类型

torch.logaddexp2

是

不支持double数据类型

torch.logical_and

是

支持bf16，fp32，fp16，uint8，int8，int16，int32，int64，bool，complex64，complex128

torch.logical_not

是

支持fp32，fp16，uint8，int8，int16，int32，int64，bool

torch.logical_or

是

支持bf16，fp32，fp16，uint8，int8，int16，int32，int64，bool，complex64，complex128

torch.logical_xor

是

支持fp32，fp16，uint8，int8，int16，int32，int64，bool，complex64，complex128

torch.logit

torch.hypot

torch.i0

torch.igamma

torch.igammac

torch.mul

是

支持bf16，fp32，fp16，uint8，int8，int16，int32，int64，bool，complex64，complex128

torch.multiply

是

支持bf16，fp32，fp16，uint8，int8，int16，int32，int64，bool，complex64，complex128

torch.mvlgamma

torch.nan_to_num

是

支持bf16，fp32，fp16，uint8，int8，int16，int32，int64，bool

torch.neg

是

支持bf16，fp32，fp16，int8，int32，int64，complex64，complex128

torch.negative

是

支持bf16，fp32，fp16，int8，int32，int64，complex64，complex128

torch.nextafter

torch.polygamma

torch.positive

是

支持bf16，fp32，fp16，uint8，int8，int16，int32，int64，complex64，complex128

torch.pow

torch.quantized_batch_norm

torch.quantized_max_pool1d

torch.quantized_max_pool2d

torch.rad2deg

是

支持fp16，fp32，fp64，int8，uint8，int16，int32，int64，bool

torch.real

是

支持fp32，fp16，complex64，complex128

torch.reciprocal

是

支持bf16，fp32，fp16，uint8，int8，int16，int32，int64，bool，complex64，complex128

torch.remainder

是

支持fp32，fp16，int16，int32，int64

torch.round

是

支持fp32，fp16

torch.rsqrt

是

支持bf16，fp32，fp16，uint8，int8，int16，int32，int64，bool，complex64，complex128

torch.sigmoid

是

支持fp32，fp16，uint8，int8，int16，int32，int64，bool，complex64，complex128

torch.sign

是

支持bf16，fp32，fp16，int32，int64，bool

torch.sgn

是

支持bf16，fp32，fp16，int32，int64，bool，complex64，complex128

torch.signbit

torch.sin

是

支持bf16，fp32，fp16，uint8，int8，int16，int32，int64，bool，complex64，complex128

torch.sinc

torch.sinh

是

支持fp16，fp32，fp64

torch.softmax

是

支持fp32

支持Named Tensor

torch.sqrt

是

支持bf16，fp16，fp32，fp64，int8，uint8，int16，int32，int64，bool

torch.square

是

支持fp16，fp32，fp64，int8，uint8，int16，int32，int64

torch.sub

是

支持bf16，fp16，fp32，fp64，uint8，int8，int16，int32，int64

torch.subtract

torch.tan

是

支持fp16，fp32，fp64

torch.tanh

是

支持bf16，fp16，fp32，int8，uint8，int16，int32，int64，bool

torch.true_divide

是

支持bf16，fp32，fp16，uint8，int8，int16，int32，int64，bool

torch.trunc

是

支持fp32，fp16

torch.xlogy

是

支持fp16，fp32，uint8，int8，int16，int32，int64，bool

torch.argmax

是

支持bf16，fp16，fp32，fp64，uint8，int8，int16，int32，int64

torch.argmin

是

支持fp32，fp16，uint8，int8，int16，int32，int64

torch.amax

是

支持fp32，fp16，uint8，int8，int16，int32，int64，bool

torch.amin

是

支持fp32，fp16，uint8，int8，int16，int32，int64，bool

torch.aminmax

是

支持fp32，fp16，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，fp32，fp16，int64，bool

torch.min

是

支持bf16，fp32，fp16，int64，bool

torch.dist

是

支持bf16，fp32，fp16

torch.logsumexp

是

支持fp32，uint8，int8，int16，int32，int64，bool，fp16

torch.mean

是

支持bf16，fp32，fp16，complex64，complex128

torch.nanmean

torch.median

是

支持bf16，fp16，fp32，int8，uint8，int16，int32，int64

torch.nanmedian

torch.mode

torch.norm

是

支持bf16，fp32，fp16，complex128

参数p为负数时，计算结果可能存在精度误差

参数dim指定为输入tensor中shape维度值为1的轴时，计算结果可能存在精度误差

torch.nansum

是

支持bf16，fp32，fp16，uint8，int8，int16，int32，int64，bool

torch.prod

是

支持fp32，fp16，uint8，int8，int16，int32，int64，bool，complex64，complex128

torch.quantile

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，int8，uint8，int16，int32，int64，bool

带dim场景不支持fp16

torch.unique_consecutive

否

torch.var

是

支持fp16，fp32

torch.var_mean

否

torch.count_nonzero

是

支持bf16，fp32，fp16，uint8，int8，int16，int32，int64，bool，complex64，complex128

torch.allclose

是

支持fp32，fp16，uint8，int8，int16，int32，int64，bool

torch.argsort

是

支持fp32，fp16，uint8，int8，int16，int32，int64

torch.eq

是

支持bf16，fp32，fp16，uint8，int8，int16，int32，int64，bool，complex64，complex128

torch.equal

是

支持fp32，fp16，uint8，int8，int16，int32，int64，bool

torch.ge

是

支持bf16，fp32，fp16，uint8，int8，int16，int32，int64，bool

torch.greater_equal

是

支持bf16，fp32，fp16，uint8，int8，int16，int32，int64，bool

torch.gt

是

支持bf16，fp32，fp16，uint8，int8，int16，int32，int64，bool

torch.greater

是

支持bf16，fp32，fp16，uint8，int8，int16，int32，int64，bool

torch.isclose

是

支持fp32，uint8，int8，int16，int32，int64，bool，fp16

torch.isfinite

是

支持bf16，fp16，fp32，fp64，uint8，int8，int16，int32，int64，bool

torch.isin

torch.isinf

是

支持bf16，fp32，fp16，uint8，int8，int16，int32，int64，bool，complex64，complex128

torch.isposinf

是

支持bf16，fp32，fp16，uint8，int8，int16，int32，int64，bool

torch.isneginf

是

支持bf16，fp32，fp16，uint8，int8，int16，int32，int64，bool

torch.isnan

是

支持bf16，fp32，fp16，uint8，int8，int16，int32，int64，bool，complex64，complex128

torch.isreal

torch.kthvalue

是

支持fp16，fp32和int32

torch.le

是

支持bf16，fp32，fp16，uint8，int8，int16，int32，int64，bool

torch.less_equal

是

支持fp16，fp32，int32，int8，uint8，fp64，int16，int64

torch.lt

是

支持bf16，fp32，fp16，uint8，int8，int16，int32，int64，bool

torch.less

是

支持bf16，fp32，fp16，uint8，int8，int16，int32，int64，bool

torch.maximum

是

支持bf16，fp32，fp16，uint8，int8，int16，int32，int64，bool

torch.minimum

是

支持bf16，fp32，fp16，uint8，int8，int16，int32，int64，bool

torch.fmax

torch.fmin

torch.ne

是

支持bf16，fp32，fp16，uint8，int8，int16，int32，int64，bool，complex64，complex128

torch.not_equal

是

支持bf16，fp32，fp16，uint8，int8，int16，int32，int64，bool，complex64，complex128

torch.sort

是

支持fp32，fp16，uint8，int8，int16，int32，int64

torch.topk

是

支持bf16，fp16，fp32，fp64，uint8，int8，int16，int32，int64

不支持sorted=False场景

torch.msort

是

支持bf16，fp32，fp16，uint8，int8，int16，int32，int64

torch.stft

是

若算子超时，需要用官方接口set_op_execute_time_out进行设置，调高超时阈值判断时间

torch.istft

torch.bartlett_window

torch.blackman_window

torch.hamming_window

torch.hann_window

是

支持bf16，fp16，fp32

torch.kaiser_window

torch.atleast_1d

是

支持bf16，fp16，fp32，fp64，int8，uint8，int16，int32，int64，bool

torch.atleast_2d

是

支持fp16，fp32，int32，int8，uint8，fp64，int16，int64，bool

torch.atleast_3d

是

支持fp16，fp32，int32，int8，uint8，fp64，int16，int64，bool

torch.bincount

是

支持uint8，int8，int16，int32，int64

torch.block_diag

是

支持bf16，fp32，fp16，uint8，int8，int16，int32，int64，bool，complex64，complex128

torch.broadcast_tensors

是

支持bf16，fp32，fp16，uint8，int8，int16，int32，int64，bool，complex64，complex128

torch.broadcast_to

是

支持bf16，fp16，fp32，fp64，int8，uint8，int16，int32，int64，bool

torch.broadcast_shapes

是

torch.bucketize

torch.cartesian_prod

torch.cdist

否

torch.clone

是

支持bf16，fp16，fp32，fp64，uint8，int8，int16，int32，int64，bool

torch.combinations

是

支持fp16，fp32，int32，int8，uint8，fp64，int16，int64，bool

torch.corrcoef

torch.cov

是

支持fp32

torch.cross

是

支持fp32，fp16，uint8，int8，int16，int32，int64，complex64，complex128

两个输入的shape要保持一致

torch.cummax

是

支持fp32，fp16，uint8，int8，int16，int32，int64，bool

torch.cummin

是

支持fp32，fp16，uint8，int8，int16，int32，int64，bool

输入为int32时，数值范围在[-16777216, 16777216]内

torch.cumprod

是

支持fp16，fp32，int32

torch.cumsum

是

支持fp32，fp16，uint8，int8，int16，int32，int64，bool，complex64，complex128

支持Named Tensor

torch.diag

是

支持fp32，fp16，uint8，int8，int16，int32，int64，bool，complex64

torch.diag_embed

是

支持bf16，fp16，fp32，uint8，int8，int16，int32，int64

torch.diagflat

torch.diagonal

是

支持bf16，fp32，fp16，uint8，int8，int16，int32，int64，bool，complex64，complex128

torch.diff

是

支持fp16，fp32，fp64，int8，uint8，int16，int32，int64，bool

torch.einsum

是

支持bf16，fp16，fp32，int8，uint8，int16，int32，int64，bool

torch.flatten

是

支持bf16，fp16，fp32，fp64，uint8，int8，int16，int32，int64，bool，complex64，complex128

torch.flip

是

支持fp32，fp16，uint8，int8，int16，int32，int64，bool，complex64，complex128

torch.fliplr

是

支持fp32，fp16，uint8，int8，int16，int32，int64，bool，complex64，complex128

torch.flipud

是

支持fp32，fp16，uint8，int8，int16，int32，int64，bool，complex64，complex128

torch.kron

是

不支持5维度及以上输入。

torch.rot90

torch.gcd

torch.histc

是

支持fp32，fp16

当输入tensor值处于计数区间交界时，归于左区间计数还是右区间计数可能存在误差

torch.histogram

torch.histogramdd

torch.meshgrid

是

支持bf16，fp16，fp32，fp64，uint8，int8，int16，int32，int64，bool

torch.lcm

torch.logcumsumexp

torch.ravel

是

支持bf16，fp32，fp16，uint8，int8，int16，int32，int64，bool，complex64，complex128

torch.renorm

torch.repeat_interleave

是

支持fp32，fp16，int16，int32，int64，bool

输入张量在重复后得到输出，输出中元素个数需小于2²²

torch.roll

是

支持fp32，fp16，int32，int64，bool

torch.searchsorted

是

支持fp16，fp32，int32，int64，fp64，int8，uint8，int16

torch.tensordot

是

支持fp32，fp16

torch.trace

torch.tril

是

支持bf16，fp32，fp16，uint8，int8，int16，int32，int64，bool

torch.tril_indices

是

torch.triu

是

支持bf16，fp32，fp16，uint8，int8，int16，int32，int64，bool

torch.triu_indices

是

torch.unflatten

是

torch.vander

torch.view_as_real

是

支持complex64

torch.view_as_complex

torch.resolve_conj

是

支持fp16，fp32，fp64，int8，uint8，int16，int32，int64，bool

torch.resolve_neg

是

支持bf16，fp32，fp16，uint8，int8，int16，int32，int64，bool，complex64，complex128

torch.addbmm

是

支持fp32，fp16

torch.addmm

是

支持fp32，fp16

torch.addmv

是

支持fp32，fp16，uint8，int8，int16，int32，int64

torch.addr

是

支持fp32，fp16，uint8，int8，int16，int32，int64，bool

torch.baddbmm

是

支持fp32，fp16

torch.bmm

是

支持fp32，fp16

torch.chain_matmul

torch.cholesky

torch.cholesky_inverse

torch.cholesky_solve

torch.dot

是

支持fp32，fp16

torch.geqrf

torch.ger

torch.inner

torch.inverse

torch.det

torch.logdet

torch.slogdet

是

支持fp32，complex64，complex128

torch.lu

torch.lu_solve

torch.lu_unpack

torch.matmul

是

支持fp32，fp16

支持Named Tensor

torch.matrix_power

torch.matrix_exp

torch.mm

是

支持fp32，fp16

torch.mv

torch.orgqr

torch.ormqr

torch.outer

是

支持bf16，fp16，fp32，fp64，uint8，int8，int16，int32，int64，bool

torch.pinverse

torch.qr

torch.svd

torch.svd_lowrank

torch.pca_lowrank

torch.lobpcg

torch.trapz

torch.trapezoid

是

支持bf16，fp32，fp16，uint8，int8，int16，int32，int64

torch.cumulative_trapezoid

是

支持fp16，fp32，fp64，int8，uint8，int16，int32，int64

torch.triangular_solve

torch.vdot

是

支持fp32，fp16

torch._foreach_abs

torch._foreach_abs_

torch._foreach_acos

torch._foreach_acos_

torch._foreach_asin

torch._foreach_asin_

torch._foreach_atan

torch._foreach_atan_

torch._foreach_ceil

torch._foreach_ceil_

torch._foreach_cos

torch._foreach_cos_

torch._foreach_cosh

torch._foreach_cosh_

torch._foreach_erf

torch._foreach_erf_

torch._foreach_erfc

torch._foreach_erfc_

torch._foreach_exp

torch._foreach_exp_

torch._foreach_expm1

torch._foreach_expm1_

torch._foreach_floor

torch._foreach_floor_

torch._foreach_log

torch._foreach_log_

torch._foreach_log10

torch._foreach_log10_

torch._foreach_log1p

torch._foreach_log1p_

torch._foreach_log2

torch._foreach_log2_

torch._foreach_neg

torch._foreach_neg_

torch._foreach_tan

torch._foreach_tan_

torch._foreach_sin

torch._foreach_sin_

torch._foreach_sinh

torch._foreach_sinh_

torch._foreach_round

torch._foreach_round_

torch._foreach_sqrt

torch._foreach_sqrt_

torch._foreach_lgamma

torch._foreach_lgamma_

torch._foreach_frac

torch._foreach_frac_

torch._foreach_reciprocal

torch._foreach_reciprocal_

torch._foreach_sigmoid

torch._foreach_sigmoid_

torch._foreach_trunc

torch._foreach_trunc_

torch._foreach_zero_

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.is_deterministic_algorithms_warn_only_enabled

torch.set_deterministic_debug_mode

是

torch.get_deterministic_debug_mode

是

torch.set_float32_matmul_precision

torch.get_float32_matmul_precision

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.sym_ite

torch.cond

torch.compile

是