下载
中文
注册

torch.Tensor

若API“是否支持”“是”“限制与说明”为空,说明此API和原生API支持度保持一致。

API名称

是否支持

限制与说明

torch.Tensor

支持fp32

Tensor.T

支持fp16,fp32,int64

Tensor.new_full

支持int64

Tensor.new_empty

支持fp32

Tensor.new_ones

支持fp32

Tensor.new_zeros

支持fp32

Tensor.is_cuda

  

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,fp16,fp32,uint8,int8,int16,int32,int64

Tensor.absolute

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

Tensor.absolute_

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

Tensor.acos

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

Tensor.acos_

支持fp16,fp32

Tensor.arccos

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

Tensor.arccos_

支持fp16,fp32

Tensor.add

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

Tensor.add_

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

Tensor.addbmm

支持fp16,fp32

Tensor.addbmm_

支持fp16,fp32

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

支持fp16,fp32

Tensor.addmm_

支持fp16,fp32

Tensor.addmv

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

Tensor.addmv_

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

Tensor.addr

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

Tensor.addr_

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

Tensor.allclose

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

Tensor.amax

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

Tensor.amin

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

Tensor.aminmax

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

Tensor.angle

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

Tensor.apply_

  

Tensor.argmax

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

Tensor.argmin

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

Tensor.argsort

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

Tensor.argwhere

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

Tensor.asin

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

Tensor.asin_

支持fp16,fp32

Tensor.arcsin

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

Tensor.arcsin_

支持fp16,fp32

Tensor.as_strided

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

Tensor.atan

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

Tensor.atan_

支持fp16,fp32

Tensor.arctan

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

Tensor.arctan_

支持fp16,fp32

Tensor.atan2

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

Tensor.atan2_

支持fp16,fp32

Tensor.arctan2

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

Tensor.arctan2_

支持fp16,fp32

Tensor.all

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

Tensor.any

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

Tensor.backward

支持fp32

Tensor.baddbmm

支持fp16,fp32

Tensor.baddbmm_

支持fp16,fp32

Tensor.bernoulli

支持fp16,fp32

Tensor.bernoulli_

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

Tensor.bfloat16

支持fp16,fp32

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

支持fp16,fp32

Tensor.bool

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

Tensor.byte

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

Tensor.broadcast_to

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

Tensor.ceil

支持fp16,fp32

Tensor.ceil_

支持fp16,fp32

Tensor.char

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

Tensor.chunk

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

Tensor.clamp

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

Tensor.clamp_

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

Tensor.clip

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

Tensor.clip_

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

Tensor.clone

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

Tensor.contiguous

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

Tensor.copy_

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

int16不支持5维以上

Tensor.conj

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

Tensor.resolve_conj

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

Tensor.resolve_neg

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

Tensor.copysign

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

Tensor.cos

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

Tensor.cos_

支持bf16,fp16,fp32,complex64,complex128

Tensor.cosh

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

Tensor.cosh_

支持bf16,fp16,fp32,fp64,complex64,complex128

Tensor.count_nonzero

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

Tensor.cov

支持fp16,fp32,int16,int32,int64

Tensor.acosh

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

Tensor.acosh_

支持bf16,fp16,fp32,fp64,complex64,complex128

Tensor.arccosh

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

Tensor.arccosh_

支持bf16,fp16,fp32,fp64,complex64,complex128

Tensor.cpu

  

Tensor.cross

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

两个输入的shape要保持一致

Tensor.cuda

  

Tensor.cummax

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

Tensor.cummin

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

Tensor.cumsum

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

支持Named Tensor

Tensor.cumsum_

支持fp16,fp32,int64,bool

Tensor.cfloat

  

Tensor.cdouble

  

Tensor.data_ptr

支持fp32

Tensor.deg2rad

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

Tensor.dequantize

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

Tensor.detach

支持fp32

Tensor.detach_

支持fp32

Tensor.diag

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

Tensor.diag_embed

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

Tensor.diagflat

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

Tensor.diagonal

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

Tensor.diagonal_scatter

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

Tensor.fill_diagonal_

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

Tensor.diff

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

Tensor.dim

支持fp32

Tensor.dim_order

  

Tensor.dist

  

Tensor.div

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

Tensor.div_

支持bf16,fp16,fp32

Tensor.divide

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

Tensor.divide_

支持fp16,fp32

Tensor.dot

支持fp16,fp32

Tensor.double

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

Tensor.dsplit

支持fp32

Tensor.element_size

支持fp32

Tensor.eq

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

Tensor.eq_

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

Tensor.equal

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

Tensor.erf

支持fp16,fp32,int64,bool

Tensor.erf_

支持fp16,fp32

Tensor.erfc

支持fp16,fp32,int64,bool

Tensor.erfc_

支持fp16,fp32

Tensor.erfinv

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

Tensor.erfinv_

支持fp16,fp32

Tensor.exp

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

Tensor.exp_

支持bf16,fp16,fp32,complex64,complex128

Tensor.expm1

支持fp16,fp32,int64,bool

Tensor.expm1_

支持fp16,fp32

Tensor.expand

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

Tensor.expand_as

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

Tensor.fix

支持fp16,fp32

Tensor.fix_

支持fp16,fp32

Tensor.fill_

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

Tensor.flatten

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

Tensor.flip

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

Tensor.fliplr

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

Tensor.flipud

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

Tensor.float

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

Tensor.float_power

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

Tensor.float_power_

支持double

Tensor.floor

支持fp16,fp32

Tensor.floor_

支持bf16,fp16,fp32

Tensor.floor_divide

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

Tensor.floor_divide_

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

Tensor.fmod

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

Tensor.fmod_

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

Tensor.frac

支持fp16,fp32

Tensor.frac_

支持fp16,fp32

Tensor.gather

支持fp16,fp32,int64

Tensor.ge

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

Tensor.ge_

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

Tensor.greater_equal

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

Tensor.greater_equal_

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

Tensor.geometric_

  

Tensor.gt

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

Tensor.gt_

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

Tensor.greater

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

Tensor.greater_

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

Tensor.half

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

Tensor.hardshrink

支持fp16,fp32

Tensor.histc

支持fp16,fp32

Tensor.hsplit

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

Tensor.index_add_

支持fp16,fp32,int64,bool

Tensor.index_add

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

Tensor.index_copy_

支持fp16,fp32,int16,int32,bool

Tensor.index_copy

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

Tensor.index_fill_

支持fp16,fp32,int32,int64,bool

Tensor.index_fill

支持fp16,fp32,int32,bool

Tensor.index_put_

支持int64

Tensor.index_put

支持int64

Tensor.index_select

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

Tensor.inner

支持fp16,fp32

Tensor.int

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

Tensor.isclose

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

Tensor.isfinite

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

Tensor.isinf

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

Tensor.isposinf

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

Tensor.isneginf

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

Tensor.isnan

支持fp32

Tensor.is_contiguous

支持fp32

Tensor.is_complex

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

Tensor.is_conj

支持fp16,fp32,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_signed

支持fp32

Tensor.is_sparse

支持fp32

Tensor.isreal

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

Tensor.item

支持fp32

Tensor.kthvalue

支持fp16,fp32和int32

Tensor.ldexp

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

Tensor.ldexp_

支持fp16,fp32,complex64,complex128

Tensor.le

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

Tensor.le_

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

Tensor.less_equal

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

Tensor.less_equal_

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

Tensor.lerp

支持fp16,fp32

Tensor.lerp_

支持fp16,fp32

Tensor.log

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

Tensor.log_

支持bf16,fp16,fp32,complex64,complex128

Tensor.log10

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

Tensor.log10_

支持bf16,fp16,fp32,fp64,complex64,complex128

Tensor.log1p

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

Tensor.log1p_

支持fp16,fp32

Tensor.log2

支持bf16,fp16,fp32,int64,bool

Tensor.log2_

支持fp16,fp32,complex64,complex128

Tensor.logaddexp

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

Tensor.logaddexp2

支持fp16,fp32

Tensor.logsumexp

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

Tensor.logical_and

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

Tensor.logical_and_

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

Tensor.logical_not

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

Tensor.logical_not_

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

Tensor.logical_or

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

Tensor.logical_or_

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

Tensor.logical_xor

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

Tensor.logical_xor_

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

Tensor.long

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

Tensor.lt

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

Tensor.lt_

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

Tensor.less

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

Tensor.less_

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

Tensor.map_

支持fp32

Tensor.masked_scatter_

支持fp32,int64,bool

Tensor.masked_scatter

支持fp32,bool

Tensor.masked_fill_

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

Tensor.masked_fill

支持fp16,fp32,int64,bool

Tensor.masked_select

支持fp32,bool

Tensor.matmul

支持bf16,fp16,fp32

支持Named Tensor

Tensor.matrix_power

支持fp16,fp32

Tensor.max

支持bf16,fp16,fp32,int64,bool

Tensor.maximum

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

Tensor.mean

支持bf16,fp16,fp32,complex64,complex128

Tensor.nanmean

支持fp16,fp32

Tensor.median

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

Tensor.min

支持bf16,fp16,fp32,int64,bool

Tensor.minimum

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

Tensor.mm

支持fp16,fp32

Tensor.movedim

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

Tensor.moveaxis

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

Tensor.msort

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

Tensor.mul

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

Tensor.mul_

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

Tensor.multiply

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

Tensor.multiply_

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

Tensor.multinomial

支持fp16,fp32

Tensor.nansum

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

Tensor.narrow

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

Tensor.narrow_copy

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

Tensor.ndimension

支持fp32

Tensor.ne

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

Tensor.ne_

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

Tensor.not_equal

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

Tensor.not_equal_

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

Tensor.neg

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

Tensor.neg_

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

Tensor.negative

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

Tensor.negative_

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

Tensor.nelement

支持fp32

Tensor.nonzero

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

不支持nan场景

Tensor.norm

支持fp16,fp32

Tensor.normal_

支持bf16,fp16,fp32

Tensor.numel

支持fp32

Tensor.numpy

支持fp32

Tensor.outer

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

Tensor.permute

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

Tensor.positive

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

Tensor.pow

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

Tensor.pow_

支持bf16,fp16,fp32,fp64,int64

Tensor.prod

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

Tensor.put_

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

Tensor.rad2deg

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

Tensor.random_

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

Tensor.ravel

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

Tensor.reciprocal

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

Tensor.reciprocal_

支持fp16,fp32,complex64,complex128

Tensor.record_stream

支持fp32

Tensor.register_hook

支持fp32

Tensor.register_post_accumulate_grad_hook

  

Tensor.remainder

支持fp16,fp32,int32,int64

Tensor.remainder_

支持fp16,fp32,int32,int64

Tensor.repeat

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

Tensor.repeat_interleave

支持fp16,fp32,int16,int32,bool

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

Tensor.requires_grad

  

Tensor.requires_grad_

支持fp32

Tensor.reshape

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

Tensor.reshape_as

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

Tensor.resize_

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

不支持传入memory_format参数

Tensor.resize_as_

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

不支持传入memory_format参数

Tensor.retain_grad

支持fp32

Tensor.retains_grad

支持fp32

Tensor.roll

支持fp16,fp32,int32,int64,bool

Tensor.rot90

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

Tensor.round

支持fp16,fp32

Tensor.round_

支持fp16,fp32

Tensor.rsqrt

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

Tensor.rsqrt_

支持fp16,fp32,complex64,complex128

Tensor.scatter

支持fp16,fp32,int16,int32,bool

Tensor.scatter_

支持fp16,fp32,int16,int32,bool

Tensor.scatter_add_

支持fp32

Tensor.scatter_add

支持fp32

Tensor.scatter_reduce

支持fp32,int64

Tensor.select

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

Tensor.select_scatter

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

Tensor.set_

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

Tensor.short

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

Tensor.sigmoid_

支持bf16,fp16,fp32,complex64,complex128

Tensor.sign

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

Tensor.sign_

支持fp16,fp32,int32,int64,bool

Tensor.sgn

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

Tensor.sgn_

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

Tensor.sin

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

Tensor.sin_

支持bf16,fp16,fp32,complex64,complex128

Tensor.sinh

支持fp16,fp32,fp64

Tensor.sinh_

支持fp16,fp32,fp64

Tensor.asinh

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

Tensor.asinh_

支持fp16,fp32,complex64,complex128

Tensor.arcsinh

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

Tensor.arcsinh_

支持fp16,fp32,complex64,complex128

Tensor.shape

  

Tensor.size

支持fp32

Tensor.slogdet

支持fp32,complex64,complex128

Tensor.slice_scatter

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

Tensor.sort

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

Tensor.split

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

Tensor.sqrt

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

Tensor.sqrt_

支持fp16,fp32,fp64

Tensor.square

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

Tensor.square_

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

Tensor.squeeze

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

Tensor.squeeze_

支持fp32

Tensor.std

支持fp16,fp32

Tensor.storage

支持fp32

Tensor.untyped_storage

支持fp32

Tensor.storage_offset

支持fp32

Tensor.storage_type

支持fp32

Tensor.stride

支持fp32

Tensor.sub

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

Tensor.sub_

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

Tensor.subtract_

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

Tensor.sum

支持bf16,fp16,fp32,int32

Tensor.sum_to_size

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

Tensor.swapaxes

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

Tensor.swapdims

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

Tensor.t

支持fp16,fp32,int64

Tensor.t_

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

Tensor.tensor_split

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

Tensor.tile

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

Tensor.to

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

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

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

to

  

Tensor.take

支持fp16,fp32,int16,int32,bool

Tensor.take_along_dim

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

Tensor.tan

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

取值范围[-65504,65504]

Tensor.tan_

支持fp16,fp32,complex64,complex128

Tensor.tanh

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

Tensor.tanh_

支持fp16,fp32

Tensor.atanh

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

Tensor.atanh_

支持fp16,fp32,complex64,complex128

Tensor.arctanh

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

Tensor.arctanh_

支持fp16,fp32,complex64,complex128

Tensor.tolist

支持fp32

Tensor.topk

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

不支持sorted=False场景

Tensor.transpose

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

Tensor.transpose_

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

Tensor.tril

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

Tensor.tril_

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

Tensor.triu

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

Tensor.triu_

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

Tensor.true_divide

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

Tensor.true_divide_

支持fp16,fp32

Tensor.trunc

支持fp16,fp32

Tensor.trunc_

支持fp16,fp32

Tensor.type

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

Tensor.type_as

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

Tensor.unbind

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

Tensor.unflatten

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

Tensor.unfold

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

Tensor.uniform_

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

Tensor.unique

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

带dim场景不支持fp16

Tensor.unique_consecutive

  

Tensor.unsqueeze

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

Tensor.unsqueeze_

支持fp32

Tensor.values

依赖稀疏tensor

Tensor.var

支持fp16,fp32

Tensor.view

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

view

  

Tensor.view_as

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

Tensor.vsplit

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

Tensor.where

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

不支持8维度的shape

Tensor.xlogy

支持fp16,fp32

Tensor.xlogy_

支持fp16,fp32

Tensor.zero_

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