torch.Tensor

若API未标明支持情况则代表该API暂无验证结果,待验证后更新表格。

API名称

是否支持

限制与说明

torch.Tensor

支持fp32

Tensor.T

支持fp32,fp16,int64

Tensor.H

     

Tensor.mT

     

Tensor.mH

     

Tensor.new_tensor

     

Tensor.new_full

支持int64

Tensor.new_empty

支持fp32

Tensor.new_ones

支持fp32

Tensor.new_zeros

支持fp32

Tensor.is_cuda

     

Tensor.is_quantized

     

Tensor.is_meta

     

Tensor.device

     

Tensor.grad

支持fp32

Tensor.ndim

支持fp32

Tensor.real

     

Tensor.imag

     

Tensor.nbytes

     

Tensor.itemsize

     

Tensor.abs

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

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

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

Tensor.addcdiv_

Atlas A2 训练系列产品:支持bf16,fp16,fp32,fp64

Atlas 训练系列产品:支持fp16,fp32,fp64

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

Tensor.addcmul

支持fp16,fp32,int64

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

Tensor.addcmul_

Atlas A2 训练系列产品:支持bf16,fp16,fp32,fp64,uint8,int8,int32,int64

Atlas 训练系列产品:支持fp16,fp32,fp64,uint8,int8,int32,int64

在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

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

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

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

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

Tensor.atan2_

支持fp32,fp16

Tensor.arctan2

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

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_

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

int16不支持5维以上

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,fp16,fp32,fp64,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

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

两个输入的shape要保持一致

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

支持Named Tensor

Tensor.cumsum_

支持fp16,fp32,int64,bool

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

     

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

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

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

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

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_leaf

     

Tensor.is_pinned

支持fp32

Tensor.is_set_to

支持fp32

Tensor.is_shared

     

Tensor.is_signed

支持fp32

Tensor.is_sparse

支持fp32

Tensor.istft

     

Tensor.isreal

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

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

支持Named Tensor

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

支持fp32,fp16

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

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

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_

支持bf16,fp16,fp32,fp64,int64

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

支持fp32

Tensor.register_hook

支持fp32

Tensor.register_post_accumulate_grad_hook

     

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

输入张量在重复后得到输出,输出中元素个数需小于222

Tensor.requires_grad

     

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

不支持传入memory_format参数

Tensor.resize_as_

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

不支持传入memory_format参数

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

支持fp32,int64

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

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

Tensor.sgn_

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

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

     

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

支持fp32

Tensor.untyped_storage

支持fp32

Tensor.storage_offset

支持fp32

Tensor.storage_type

支持fp32

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

当前NPU设备仅支持设置memory_format为torch.contiguous_format或torch.preserve_format

Atlas 推理系列产品不支持跨NPU拷贝

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

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

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_

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

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

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

Tensor.unfold

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

Tensor.uniform_

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

Tensor.unique

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

带dim场景不支持fp16

Tensor.unique_consecutive

  

Tensor.unsqueeze

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

Tensor.unsqueeze_

支持fp32

Tensor.values

依赖稀疏tensor

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