torch.Tensor

若API未标明支持情况,则代表该API的支持情况待验证。

API名称

是否支持

限制与说明

torch.Tensor

是

支持fp32

Tensor.T

是

支持fp32,fp16,int64

Tensor.new_tensor

是

  

Tensor.new_full

是

支持int64

Tensor.new_empty

是

支持fp32

Tensor.new_ones

是

支持fp32

Tensor.new_zeros

是

支持fp32

Tensor.grad

是

支持fp32

Tensor.ndim

是

支持fp32

Tensor.abs

是

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

Tensor.abs_

是

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

Tensor.absolute

是

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

Tensor.absolute_

是

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

Tensor.acos

是

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

Tensor.acos_

是

支持fp32,fp16

Tensor.arccos

是

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

Tensor.arccos_

是

支持fp32,fp16

Tensor.add

是

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

Tensor.add_

是

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

Tensor.addbmm

是

支持fp32,fp16

Tensor.addbmm_

是

支持fp32,fp16

Tensor.addcdiv

是

支持fp16,fp32,int64,bool

在int64类型不支持三个tensor同时广播

Tensor.addcdiv_

是

支持fp16,fp32,int64,bool

在int64类型不支持三个tensor同时广播

Tensor.addcmul

是

支持fp16,fp32,int64,bool

在int64类型不支持三个tensor同时广播

Tensor.addcmul_

是

支持fp16,fp32,int64,bool

在int64类型不支持三个tensor同时广播

Tensor.addmm

是

支持fp32,fp16

Tensor.addmm_

是

支持fp32,fp16

Tensor.sspaddmm

     

Tensor.addmv

是

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

Tensor.addmv_

是

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

Tensor.addr

是

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

Tensor.addr_

是

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

Tensor.adjoint

否

  

Tensor.allclose

是

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

Tensor.amax

是

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

Tensor.amin

是

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

Tensor.aminmax

是

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

Tensor.angle

否

  

Tensor.apply_

是

  

Tensor.argmax

是

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

Tensor.argmin

是

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

Tensor.argsort

是

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

Tensor.argwhere

是

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

Tensor.asin

     

Tensor.asin_

是

支持fp32,fp16

Tensor.arcsin

是

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

Tensor.arcsin_

是

支持fp32,fp16

Tensor.as_strided

是

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

Tensor.atan

是

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

Tensor.atan_

是

支持fp32,fp16

Tensor.arctan

是

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

Tensor.arctan_

是

支持fp32,fp16

Tensor.atan2

     

Tensor.atan2_

是

支持fp32,fp16

Tensor.arctan2

     

Tensor.arctan2_

是

支持fp32,fp16

Tensor.all

是

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

Tensor.any

是

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

Tensor.backward

是

支持fp32

Tensor.baddbmm

是

支持fp32,fp16

Tensor.baddbmm_

是

支持fp32,fp16

Tensor.bernoulli

是

支持fp32,fp16

Tensor.bernoulli_

是

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

Tensor.bfloat16

是

支持fp32,fp16

Tensor.bincount

是

支持uint8,int8,int16,int32,int64

Tensor.bitwise_not

是

支持uint8,int8,int16,int32,int64,bool

Tensor.bitwise_not_

是

支持uint8,int8,int16,int32,int64,bool

Tensor.bitwise_and

是

支持uint8,int8,int16,int32,int64,bool

Tensor.bitwise_and_

是

支持uint8,int8,int16,int32,int64,bool

Tensor.bitwise_or

是

支持uint8,int8,int16,int32,int64,bool

Tensor.bitwise_or_

是

支持uint8,int8,int16,int32,int64,bool

Tensor.bitwise_xor

是

支持uint8,int8,int16,int32,int64,bool

Tensor.bitwise_xor_

是

支持uint8,int8,int16,int32,int64,bool

Tensor.bitwise_left_shift

否

  

Tensor.bitwise_left_shift_

     

Tensor.bitwise_right_shift

否

  

Tensor.bitwise_right_shift_

     

Tensor.bmm

是

支持fp32,fp16

Tensor.bool

是

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

Tensor.byte

是

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

Tensor.broadcast_to

是

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

Tensor.cauchy_

     

Tensor.ceil

是

支持fp32,fp16

Tensor.ceil_

是

支持fp32,fp16

Tensor.char

是

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

Tensor.cholesky

否

  

Tensor.cholesky_inverse

否

  

Tensor.cholesky_solve

否

  

Tensor.chunk

是

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

Tensor.clamp

是

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

Tensor.clamp_

是

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

Tensor.clip

是

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

Tensor.clip_

是

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

Tensor.clone

是

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

Tensor.contiguous

是

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

Tensor.copy_

是

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

Tensor.conj

是

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

Tensor.conj_physical

否

  

Tensor.conj_physical_

     

Tensor.resolve_conj

是

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

Tensor.resolve_neg

是

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

Tensor.copysign

是

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

Tensor.copysign_

     

Tensor.cos

是

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

Tensor.cos_

是

支持bf16,fp32,fp16,complex64,complex128

Tensor.cosh

是

支持fp16,fp32,fp64

Tensor.cosh_

是

支持fp16,fp32,fp64

Tensor.corrcoef

     

Tensor.count_nonzero

是

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

Tensor.cov

是

支持fp32,fp16,int16,int32,int64

Tensor.acosh

是

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

Tensor.acosh_

是

支持fp16,fp32,fp64

Tensor.arccosh

是

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

Tensor.arccosh_

是

支持fp16,fp32,fp64

Tensor.cpu

是

  

Tensor.cross

     

Tensor.cuda

     

Tensor.logcumsumexp

否

  

Tensor.cummax

是

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

Tensor.cummin

是

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

Tensor.cumprod

     

Tensor.cumprod_

     

Tensor.cumsum

是

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

Tensor.cumsum_

     

Tensor.chalf

否

  

Tensor.cfloat

否

  

Tensor.cdouble

否

  

Tensor.data_ptr

是

支持fp32

Tensor.deg2rad

是

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

Tensor.dequantize

是

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

Tensor.det

否

  

Tensor.dense_dim

     

Tensor.detach

是

支持fp32

Tensor.detach_

是

支持fp32

Tensor.diag

是

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

Tensor.diag_embed

是

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

Tensor.diagflat

是

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

Tensor.diagonal

是

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

Tensor.diagonal_scatter

是

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

Tensor.fill_diagonal_

是

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

Tensor.fmax

否

  

Tensor.fmin

否

  

Tensor.diff

是

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

Tensor.digamma

否

  

Tensor.digamma_

     

Tensor.dim

是

支持fp32

Tensor.dist

     

Tensor.div

是

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

Tensor.div_

是

支持bf16,fp32,fp16

Tensor.divide

是

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

Tensor.divide_

是

支持fp32,fp16

Tensor.dot

是

支持fp32,fp16

Tensor.double

是

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

Tensor.dsplit

是

支持fp32

Tensor.element_size

是

支持fp32

Tensor.eq

是

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

Tensor.eq_

是

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

Tensor.equal

是

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

Tensor.erf

是

支持fp32,fp16,int64,bool

Tensor.erf_

是

支持fp32,fp16

Tensor.erfc

是

支持fp32,fp16,int64,bool

Tensor.erfc_

是

支持fp32,fp16

Tensor.erfinv

是

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

Tensor.erfinv_

是

支持fp32,fp16

Tensor.exp

是

支持bf16,fp32,fp16,int64,bool,complex64,complex128

Tensor.exp_

是

支持bf16,fp32,fp16,complex64,complex128

Tensor.expm1

是

支持fp32,fp16,int64,bool

Tensor.expm1_

是

支持fp32,fp16

Tensor.expand

是

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

Tensor.expand_as

是

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

Tensor.exponential_

否

  

Tensor.fix

是

支持fp32,fp16

Tensor.fix_

是

支持fp32,fp16

Tensor.fill_

是

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

Tensor.flatten

是

支持fp32

Tensor.flip

是

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

Tensor.fliplr

是

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

Tensor.flipud

是

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

Tensor.float

是

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

Tensor.float_power

是

支持fp32,fp16,uint8,int8,int16,int32,int64,bool,complex128

Tensor.float_power_

是

支持double

Tensor.floor

是

支持fp32,fp16

Tensor.floor_

是

支持bf16,fp32,fp16

Tensor.floor_divide

是

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

Tensor.floor_divide_

是

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

Tensor.fmod

是

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

Tensor.fmod_

是

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

Tensor.frac

是

支持fp32,fp16

Tensor.frac_

是

支持fp32,fp16

Tensor.frexp

否

  

Tensor.gather

是

支持fp32,fp16,int64

Tensor.gcd

否

  

Tensor.gcd_

     

Tensor.ge

是

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

Tensor.ge_

是

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

Tensor.greater_equal

是

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

Tensor.greater_equal_

是

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

Tensor.geometric_

否

  

Tensor.geqrf

否

  

Tensor.ger

     

Tensor.get_device

否

  

Tensor.gt

是

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

Tensor.gt_

是

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

Tensor.greater

是

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

Tensor.greater_

是

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

Tensor.half

是

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

Tensor.hardshrink

是

支持fp32,fp16

Tensor.heaviside

否

  

Tensor.histc

是

支持fp32,fp16

Tensor.histogram

否

  

Tensor.hsplit

是

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

Tensor.hypot

否

  

Tensor.hypot_

     

Tensor.i0

否

  

Tensor.i0_

     

Tensor.igamma

否

  

Tensor.igamma_

     

Tensor.igammac

否

  

Tensor.igammac_

     

Tensor.index_add_

是

支持fp32,fp16,int64,bool

Tensor.index_add

是

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

Tensor.index_copy_

是

支持fp32,fp16,int16,int32,bool

Tensor.index_copy

     

Tensor.index_fill_

是

支持fp32,fp16,int32,bool,int64

Tensor.index_fill

是

支持fp32,fp16,int32,bool

Tensor.index_put_

是

支持bp16,fp16,fp32,fp64,int8,uint8,int16,int32,int64,bool

Tensor.index_put

是

支持bp16,fp16,fp32,fp64,int8,uint8,int16,int32,int64,bool

Tensor.index_reduce_

否

  

Tensor.index_reduce

否

  

Tensor.index_select

是

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

Tensor.indices

否

  

Tensor.inner

是

支持fp32,fp16

Tensor.int

是

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

Tensor.int_repr

     

Tensor.inverse

     

Tensor.isclose

是

支持fp32,uint8,int32,int64,bool,fp16

Tensor.isfinite

是

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

Tensor.isinf

是

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

Tensor.isposinf

是

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

Tensor.isneginf

是

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

Tensor.isnan

是

支持fp32

Tensor.is_contiguous

是

支持fp32

Tensor.is_complex

是

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

Tensor.is_conj

是

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

Tensor.is_floating_point

是

支持fp32

Tensor.is_inference

是

支持fp32

Tensor.is_pinned

是

支持fp32

Tensor.is_set_to

否

  

Tensor.is_shared

否

  

Tensor.is_signed

是

支持fp32

Tensor.is_sparse

是

支持fp32

Tensor.istft

否

  

Tensor.isreal

     

Tensor.item

是

支持fp32

Tensor.kthvalue

是

支持fp16,fp32和int32

Tensor.lcm

否

  

Tensor.lcm_

     

Tensor.ldexp

是

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

Tensor.ldexp_

是

支持fp32,fp16,complex64,complex128

Tensor.le

是

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

Tensor.le_

是

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

Tensor.less_equal

是

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

Tensor.less_equal_

是

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

Tensor.lerp

是

支持fp32,fp16

Tensor.lerp_

是

支持fp32,fp16

Tensor.lgamma

否

  

Tensor.lgamma_

     

Tensor.log

是

支持bf16,fp32,fp16,int64,bool,complex64,complex128

Tensor.log_

是

支持bf16,fp32,fp16,complex64,complex128

Tensor.logdet

否

  

Tensor.log10

是

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

Tensor.log10_

是

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

Tensor.log1p

是

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

Tensor.log1p_

是

支持fp32,fp16

Tensor.log2

是

支持bf16,fp32,fp16,int64,bool

Tensor.log2_

是

支持fp32,fp16,complex64,complex128

Tensor.log_normal_

否

  

Tensor.logaddexp

是

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

Tensor.logaddexp2

是

支持fp32,fp16

Tensor.logsumexp

是

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

Tensor.logical_and

是

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

Tensor.logical_and_

是

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

Tensor.logical_not

是

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

Tensor.logical_not_

是

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

Tensor.logical_or

是

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

Tensor.logical_or_

是

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

Tensor.logical_xor

是

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

Tensor.logical_xor_

是

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

Tensor.logit

否

  

Tensor.logit_

     

Tensor.long

是

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

Tensor.lt

是

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

Tensor.lt_

是

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

Tensor.less

是

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

Tensor.less_

是

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

Tensor.lu

否

  

Tensor.lu_solve

否

  

Tensor.as_subclass

     

Tensor.map_

是

支持fp32

Tensor.masked_scatter_

是

支持fp32,int64,bool

Tensor.masked_scatter

是

支持fp32,bool

Tensor.masked_fill_

是

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

Tensor.masked_fill

是

支持fp32,fp16,int64,bool

Tensor.masked_select

是

支持fp32,bool

Tensor.matmul

是

支持bf16,fp16,fp32

Tensor.matrix_power

是

支持fp32,fp16

Tensor.matrix_exp

否

  

Tensor.max

是

支持bf16,fp32,fp16,int64,bool

Tensor.maximum

是

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

Tensor.mean

是

支持bf16,fp32,fp16,complex64,complex128

Tensor.nanmean

否

  

Tensor.median

是

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

Tensor.nanmedian

否

  

Tensor.min

是

支持bf16,fp32,fp16,int64,bool

Tensor.minimum

是

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

Tensor.mm

是

支持fp32,fp16

Tensor.smm

     

Tensor.mode

否

  

Tensor.movedim

是

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

Tensor.moveaxis

是

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

Tensor.msort

是

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

Tensor.mul

是

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

Tensor.mul_

是

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

Tensor.multiply

是

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

Tensor.multiply_

是

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

Tensor.multinomial

是

支持fp32,fp16

Tensor.mv

     

Tensor.mvlgamma

否

  

Tensor.mvlgamma_

     

Tensor.nansum

否

  

Tensor.narrow

是

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

Tensor.narrow_copy

是

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

Tensor.ndimension

是

支持fp32

Tensor.nan_to_num

否

  

Tensor.nan_to_num_

     

Tensor.ne

是

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

Tensor.ne_

是

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

Tensor.not_equal

是

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

Tensor.not_equal_

是

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

Tensor.neg

是

支持fp32,fp16,int8,int32,int64,complex64,complex128

Tensor.neg_

是

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

Tensor.negative

是

支持fp32,fp16,int8,int32,int64,complex64,complex128

Tensor.negative_

是

支持fp32,fp16,int8,int32,int64,complex64,complex128

Tensor.nelement

是

支持fp32

Tensor.nextafter

否

  

Tensor.nextafter_

     

Tensor.nonzero

是

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

不支持nan场景

Tensor.norm

是

支持fp32,fp16,complex128

Tensor.normal_

是

支持bf16,fp32,fp16

Tensor.numel

是

支持fp32

Tensor.numpy

是

支持fp32

Tensor.orgqr

否

  

Tensor.ormqr

否

  

Tensor.outer

是

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

Tensor.permute

是

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

Tensor.pin_memory

     

Tensor.pinverse

     

Tensor.polygamma

否

  

Tensor.polygamma_

     

Tensor.positive

是

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

Tensor.pow

是

支持bf16,,fp16,fp32,fp64,int16,int64

Tensor.pow_

     

Tensor.prod

是

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

Tensor.put_

是

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

Tensor.qr

     

Tensor.qscheme

     

Tensor.quantile

     

Tensor.nanquantile

否

  

Tensor.q_scale

     

Tensor.q_zero_point

     

Tensor.q_per_channel_scales

     

Tensor.q_per_channel_zero_points

     

Tensor.q_per_channel_axis

     

Tensor.rad2deg

是

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

Tensor.random_

是

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

Tensor.ravel

是

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

Tensor.reciprocal

是

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

Tensor.reciprocal_

是

支持fp32,fp16,complex64,complex128

Tensor.record_stream

     

Tensor.register_hook

是

支持fp32

Tensor.remainder

是

支持fp32,fp16,int32,int64

Tensor.remainder_

是

支持fp32,fp16,int32,int64

Tensor.renorm

     

Tensor.renorm_

     

Tensor.repeat

是

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

Tensor.repeat_interleave

是

支持fp32,fp16,int16,int32,bool

Tensor.requires_grad_

是

支持fp32

Tensor.reshape

是

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

Tensor.reshape_as

是

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

Tensor.resize_

是

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

Tensor.resize_as_

是

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

Tensor.retain_grad

是

支持fp32

Tensor.retains_grad

是

支持fp32

Tensor.roll

是

支持fp32,fp16,int32,int64,bool

Tensor.rot90

是

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

Tensor.round

是

支持fp32,fp16

Tensor.round_

是

支持fp32,fp16

Tensor.rsqrt

是

支持fp32,uint8,int8,int16,int32,int64,bool,complex64,complex128fp16

Tensor.rsqrt_

是

支持fp32,fp16,complex64,complex128

Tensor.scatter

是

支持fp32,fp16,int16,int32,bool

Tensor.scatter_

是

支持fp32,fp16,int16,int32,bool

Tensor.scatter_add_

是

支持fp32

Tensor.scatter_add

是

支持fp32

Tensor.scatter_reduce_

否

  

Tensor.scatter_reduce

否

  

Tensor.select

是

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

Tensor.select_scatter

是

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

Tensor.set_

是

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

Tensor.share_memory_

否

  

Tensor.short

是

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

Tensor.sigmoid

     

Tensor.sigmoid_

是

支持bf16,fp32,fp16,complex64,complex128

Tensor.sign

是

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

Tensor.sign_

是

支持fp32,fp16,int32,int64,bool

Tensor.signbit

否

  

Tensor.sgn

否

  

Tensor.sgn_

     

Tensor.sin

是

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

Tensor.sin_

是

支持bf16,fp32,fp16,complex64,complex128

Tensor.sinc

否

  

Tensor.sinc_

     

Tensor.sinh

是

支持fp16,fp32,fp64

Tensor.sinh_

是

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

Tensor.asinh

是

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

Tensor.asinh_

是

支持fp32,fp16,complex64,complex128

Tensor.arcsinh

是

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

Tensor.arcsinh_

是

支持fp32,fp16,complex64,complex128

Tensor.size

是

支持fp32

Tensor.slogdet

是

支持fp32,complex64,complex128

Tensor.slice_scatter

是

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

Tensor.softmax

否

  

Tensor.sort

是

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

Tensor.split

是

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

Tensor.sparse_mask

     

Tensor.sparse_dim

     

Tensor.sqrt

是

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

Tensor.sqrt_

是

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

Tensor.square

是

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

Tensor.square_

是

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

Tensor.squeeze

是

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

Tensor.squeeze_

是

支持fp32

Tensor.std

是

支持fp16,fp32

Tensor.stft

否

  

Tensor.storage

否

  

Tensor.untyped_storage

否

  

Tensor.storage_offset

是

支持fp32

Tensor.storage_type

否

  

Tensor.stride

是

支持fp32

Tensor.sub

是

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

Tensor.sub_

是

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

Tensor.subtract

     

Tensor.subtract_

是

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

Tensor.sum

     

Tensor.sum_to_size

是

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

Tensor.svd

     

Tensor.swapaxes

是

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

Tensor.swapdims

是

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

Tensor.t

是

支持fp32,fp16,int64

Tensor.t_

是

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

Tensor.tensor_split

是

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

Tensor.tile

是

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

Tensor.to

是

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

to

     

torch.to

     

torch.to

     

Tensor.to_mkldnn

     

Tensor.take

是

支持fp32,fp16,int16,int32,bool

Tensor.take_along_dim

是

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

Tensor.tan

是

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

Tensor.tan_

是

支持fp32,fp16,complex64,complex128

Tensor.tanh

是

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

Tensor.tanh_

是

支持fp32,fp16

Tensor.atanh

是

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

Tensor.atanh_

是

支持fp32,fp16,complex64,complex128

Tensor.arctanh

     

Tensor.arctanh_

是

支持fp32,fp16,complex64,complex128

Tensor.tolist

是

支持fp32

Tensor.topk

是

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

不支持sorted=False场景

Tensor.to_dense

     

Tensor.to_sparse

     

to_sparse

否

  

Tensor.to_sparse_csr

否

  

Tensor.to_sparse_csc

否

  

Tensor.to_sparse_bsr

否

  

Tensor.to_sparse_bsc

否

  

Tensor.trace

     

Tensor.transpose

是

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

Tensor.transpose_

     

Tensor.triangular_solve

     

Tensor.tril

是

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

Tensor.tril_

是

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

Tensor.triu

是

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

Tensor.triu_

是

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

Tensor.true_divide

是

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

Tensor.true_divide_

是

支持fp32,fp16

Tensor.trunc

是

支持fp32,fp16

Tensor.trunc_

是

支持fp32,fp16

Tensor.type

是

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

Tensor.type_as

是

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

Tensor.unbind

是

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

Tensor.unflatten

否

  

Tensor.unfold

是

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

Tensor.uniform_

是

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

Tensor.unique

     

Tensor.unique_consecutive

     

Tensor.unsqueeze

是

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

Tensor.unsqueeze_

是

支持fp32

Tensor.values

     

Tensor.var

是

支持fp16,fp32

Tensor.vdot

否

  

Tensor.view

是

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

view

     

Tensor.view_as

是

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

Tensor.vsplit

是

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

Tensor.where

是

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

不支持8维度的shape

Tensor.xlogy

是

支持fp16,fp32,fp64

Tensor.xlogy_

是

支持fp16,fp32,fp64

Tensor.zero_

是

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