torch.Tensor

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

API名称

是否支持

限制与说明

torch.Tensor

  

Tensor.T

  

Tensor.new_tensor

     

Tensor.new_full

     

Tensor.new_empty

     

Tensor.new_ones

     

Tensor.new_zeros

     

Tensor.grad

  

Tensor.ndim

  

Tensor.abs

  

Tensor.abs_

     

Tensor.absolute

     

Tensor.absolute_

     

Tensor.acos

     

Tensor.acos_

     

Tensor.arccos

     

Tensor.arccos_

     

Tensor.add

  

Tensor.add_

  

Tensor.addbmm

     

Tensor.addbmm_

     

Tensor.addcdiv

     

Tensor.addcdiv_

     

Tensor.addcmul

     

Tensor.addcmul_

     

Tensor.addmm

     

Tensor.addmm_

     

Tensor.sspaddmm

     

Tensor.addmv

     

Tensor.addmv_

     

Tensor.addr

     

Tensor.addr_

     

Tensor.adjoint

  

Tensor.allclose

     

Tensor.amax

     

Tensor.amin

     

Tensor.aminmax

  

Tensor.angle

  

Tensor.apply_

     

Tensor.argmax

  

Tensor.argmin

  

Tensor.argsort

  

Tensor.argwhere

     

Tensor.asin

     

Tensor.asin_

     

Tensor.arcsin

     

Tensor.arcsin_

     

Tensor.as_strided

     

Tensor.atan

     

Tensor.atan_

     

Tensor.arctan

     

Tensor.arctan_

     

Tensor.atan2

     

Tensor.atan2_

     

Tensor.arctan2

     

Tensor.arctan2_

     

Tensor.all

     

Tensor.any

     

Tensor.backward

  

Tensor.baddbmm

     

Tensor.baddbmm_

     

Tensor.bernoulli

     

Tensor.bernoulli_

     

Tensor.bfloat16

     

Tensor.bincount

  

Tensor.bitwise_not

     

Tensor.bitwise_not_

     

Tensor.bitwise_and

     

Tensor.bitwise_and_

     

Tensor.bitwise_or

     

Tensor.bitwise_or_

     

Tensor.bitwise_xor

     

Tensor.bitwise_xor_

     

Tensor.bitwise_left_shift

  

Tensor.bitwise_left_shift_

     

Tensor.bitwise_right_shift

  

Tensor.bitwise_right_shift_

     

Tensor.bmm

     

Tensor.bool

  

Tensor.byte

     

Tensor.broadcast_to

     

Tensor.cauchy_

     

Tensor.ceil

     

Tensor.ceil_

     

Tensor.char

     

Tensor.cholesky

  

Tensor.cholesky_inverse

  

Tensor.cholesky_solve

  

Tensor.chunk

  

Tensor.clamp

     

Tensor.clamp_

     

Tensor.clip

     

Tensor.clip_

     

Tensor.clone

  

Tensor.contiguous

  

Tensor.copy_

  

Tensor.conj

  

Tensor.conj_physical

  

Tensor.conj_physical_

     

Tensor.resolve_conj

  

Tensor.resolve_neg

  

Tensor.copysign

     

Tensor.copysign_

     

Tensor.cos

     

Tensor.cos_

     

Tensor.cosh

     

Tensor.cosh_

     

Tensor.corrcoef

     

Tensor.count_nonzero

     

Tensor.cov

     

Tensor.acosh

     

Tensor.acosh_

     

Tensor.arccosh

     

Tensor.arccosh_

     

Tensor.cpu

     

Tensor.cross

     

Tensor.cuda

     

Tensor.logcumsumexp

  

Tensor.cummax

     

Tensor.cummin

     

Tensor.cumprod

     

Tensor.cumprod_

     

Tensor.cumsum

     

Tensor.cumsum_

     

Tensor.chalf

  

Tensor.cfloat

  

Tensor.cdouble

  

Tensor.data_ptr

  

Tensor.deg2rad

     

Tensor.dequantize

  

Tensor.det

  

Tensor.dense_dim

     

Tensor.detach

  

Tensor.detach_

  

Tensor.diag

     

Tensor.diag_embed

     

Tensor.diagflat

     

Tensor.diagonal

  

Tensor.diagonal_scatter

  

Tensor.fill_diagonal_

     

Tensor.fmax

  

Tensor.fmin

  

Tensor.diff

     

Tensor.digamma

  

Tensor.digamma_

     

Tensor.dim

  

Tensor.dist

     

Tensor.div

     

Tensor.div_

     

Tensor.divide

  

Tensor.divide_

     

Tensor.dot

     

Tensor.double

  

Tensor.dsplit

  

Tensor.element_size

  

Tensor.eq

     

Tensor.eq_

     

Tensor.equal

     

Tensor.erf

     

Tensor.erf_

     

Tensor.erfc

     

Tensor.erfc_

     

Tensor.erfinv

     

Tensor.erfinv_

     

Tensor.exp

     

Tensor.exp_

     

Tensor.expm1

     

Tensor.expm1_

     

Tensor.expand

  

Tensor.expand_as

  

Tensor.exponential_

  

Tensor.fix

     

Tensor.fix_

     

Tensor.fill_

     

Tensor.flatten

  

Tensor.flip

  

Tensor.fliplr

     

Tensor.flipud

     

Tensor.float

  

Tensor.float_power

     

Tensor.float_power_

     

Tensor.floor

     

Tensor.floor_

     

Tensor.floor_divide

     

Tensor.floor_divide_

     

Tensor.fmod

     

Tensor.fmod_

     

Tensor.frac

     

Tensor.frac_

     

Tensor.frexp

  

Tensor.gather

  

Tensor.gcd

  

Tensor.gcd_

     

Tensor.ge

     

Tensor.ge_

     

Tensor.greater_equal

     

Tensor.greater_equal_

     

Tensor.geometric_

  

Tensor.geqrf

  

Tensor.ger

     

Tensor.get_device

  

Tensor.gt

  

Tensor.gt_

  

Tensor.greater

     

Tensor.greater_

     

Tensor.half

  

Tensor.hardshrink

     

Tensor.heaviside

  

Tensor.histc

     

Tensor.histogram

  

Tensor.hsplit

     

Tensor.hypot

  

Tensor.hypot_

     

Tensor.i0

  

Tensor.i0_

     

Tensor.igamma

  

Tensor.igamma_

     

Tensor.igammac

  

Tensor.igammac_

     

Tensor.index_add_

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

Tensor.index_add

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

Tensor.index_copy_

     

Tensor.index_copy

     

Tensor.index_fill_

     

Tensor.index_fill

     

Tensor.index_put_

     

Tensor.index_put

     

Tensor.index_reduce_

  

Tensor.index_reduce

  

Tensor.index_select

  

Tensor.indices

  

Tensor.inner

     

Tensor.int

  

Tensor.int_repr

     

Tensor.inverse

     

Tensor.isclose

     

Tensor.isfinite

  

Tensor.isinf

     

Tensor.isposinf

     

Tensor.isneginf

     

Tensor.isnan

  

Tensor.is_contiguous

  

Tensor.is_complex

     

Tensor.is_conj

     

Tensor.is_floating_point

  

Tensor.is_inference

  

Tensor.is_pinned

  

Tensor.is_set_to

  

Tensor.is_shared

  

Tensor.is_signed

  

Tensor.is_sparse

  

Tensor.istft

  

Tensor.isreal

     

Tensor.item

  

Tensor.kthvalue

  

Tensor.lcm

  

Tensor.lcm_

     

Tensor.ldexp

     

Tensor.ldexp_

     

Tensor.le

     

Tensor.le_

     

Tensor.less_equal

     

Tensor.less_equal_

     

Tensor.lerp

     

Tensor.lerp_

     

Tensor.lgamma

  

Tensor.lgamma_

     

Tensor.log

     

Tensor.log_

     

Tensor.logdet

  

Tensor.log10

     

Tensor.log10_

     

Tensor.log1p

     

Tensor.log1p_

     

Tensor.log2

     

Tensor.log2_

     

Tensor.log_normal_

  

Tensor.logaddexp

  

Tensor.logaddexp2

  

Tensor.logsumexp

     

Tensor.logical_and

     

Tensor.logical_and_

     

Tensor.logical_not

     

Tensor.logical_not_

     

Tensor.logical_or

     

Tensor.logical_or_

     

Tensor.logical_xor

  

Tensor.logical_xor_

     

Tensor.logit

  

Tensor.logit_

     

Tensor.long

  

Tensor.lt

  

Tensor.lt_

     

Tensor.less

     

Tensor.less_

     

Tensor.lu

  

Tensor.lu_solve

  

Tensor.as_subclass

     

Tensor.map_

只支持CPU,GPU/NPU不支持

Tensor.masked_scatter_

     

Tensor.masked_scatter

     

Tensor.masked_fill_

  

Tensor.masked_fill

  

Tensor.masked_select

     

Tensor.matmul

     

Tensor.matrix_power

     

Tensor.matrix_exp

  

Tensor.max

     

Tensor.maximum

     

Tensor.mean

  

Tensor.nanmean

  

Tensor.median

     

Tensor.nanmedian

  

Tensor.min

     

Tensor.minimum

     

Tensor.mm

     

Tensor.smm

     

Tensor.mode

  

Tensor.movedim

     

Tensor.moveaxis

     

Tensor.msort

  

Tensor.mul

     

Tensor.mul_

     

Tensor.multiply

     

Tensor.multiply_

     

Tensor.multinomial

     

Tensor.mv

     

Tensor.mvlgamma

  

Tensor.mvlgamma_

     

Tensor.nansum

  

Tensor.narrow

     

Tensor.narrow_copy

     

Tensor.ndimension

  

Tensor.nan_to_num

  

Tensor.nan_to_num_

     

Tensor.ne

  

Tensor.ne_

     

Tensor.not_equal

     

Tensor.not_equal_

     

Tensor.neg

     

Tensor.neg_

     

Tensor.negative

     

Tensor.negative_

     

Tensor.nelement

  

Tensor.nextafter

  

Tensor.nextafter_

     

Tensor.nonzero

  

Tensor.norm

     

Tensor.normal_

  

Tensor.numel

  

Tensor.numpy

     

Tensor.orgqr

  

Tensor.ormqr

  

Tensor.outer

     

Tensor.permute

  

Tensor.pin_memory

     

Tensor.pinverse

     

Tensor.polygamma

  

Tensor.polygamma_

     

Tensor.positive

     

Tensor.pow

     

Tensor.pow_

     

Tensor.prod

     

Tensor.put_

     

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

     

Tensor.random_

     

Tensor.ravel

  

Tensor.reciprocal

     

Tensor.reciprocal_

     

Tensor.record_stream

     

Tensor.register_hook

  

Tensor.remainder

  

Tensor.remainder_

     

Tensor.renorm

支持fp16,fp32。

Tensor.renorm_

支持fp16,fp32。

Tensor.repeat

     

Tensor.repeat_interleave

  

Tensor.requires_grad_

  

Tensor.reshape

  

Tensor.reshape_as

  

Tensor.resize_

  

Tensor.resize_as_

     

Tensor.retain_grad

  

Tensor.retains_grad

  

Tensor.roll

     

Tensor.rot90

     

Tensor.round

     

Tensor.round_

     

Tensor.rsqrt

     

Tensor.rsqrt_

     

Tensor.scatter

     

Tensor.scatter_

     

Tensor.scatter_add_

     

Tensor.scatter_add

     

Tensor.scatter_reduce_

  

Tensor.scatter_reduce

  

Tensor.select

  

Tensor.select_scatter

     

Tensor.set_

     

Tensor.share_memory_

  

Tensor.short

     

Tensor.sigmoid

     

Tensor.sigmoid_

     

Tensor.sign

     

Tensor.sign_

     

Tensor.signbit

  

Tensor.sgn

  

Tensor.sgn_

     

Tensor.sin

     

Tensor.sin_

     

Tensor.sinc

  

Tensor.sinc_

     

Tensor.sinh

     

Tensor.sinh_

     

Tensor.asinh

     

Tensor.asinh_

     

Tensor.arcsinh

     

Tensor.arcsinh_

     

Tensor.size

  

Tensor.slogdet

     

Tensor.slice_scatter

     

Tensor.softmax

  

Tensor.sort

     

Tensor.split

  

Tensor.sparse_mask

     

Tensor.sparse_dim

     

Tensor.sqrt

     

Tensor.sqrt_

     

Tensor.square

     

Tensor.square_

     

Tensor.squeeze

  

Tensor.squeeze_

  

Tensor.std

  

Tensor.stft

  

Tensor.storage

  

Tensor.untyped_storage

  

Tensor.storage_offset

  

Tensor.storage_type

  

Tensor.stride

  

Tensor.sub

     

Tensor.sub_

     

Tensor.subtract

     

Tensor.subtract_

     

Tensor.sum

     

Tensor.sum_to_size

     

Tensor.svd

     

Tensor.swapaxes

  

Tensor.swapdims

     

Tensor.t

  

Tensor.t_

  

Tensor.tensor_split

  

Tensor.tile

     

Tensor.to

     

to

     

torch.to

     

torch.to

     

Tensor.to_mkldnn

     

Tensor.take

     

Tensor.take_along_dim

     

Tensor.tan

     

Tensor.tan_

     

Tensor.tanh

     

Tensor.tanh_

     

Tensor.atanh

     

Tensor.atanh_

     

Tensor.arctanh

     

Tensor.arctanh_

     

Tensor.tolist

  

Tensor.topk

  

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

  

Tensor.transpose_

     

Tensor.triangular_solve

     

Tensor.tril

     

Tensor.tril_

     

Tensor.triu

  

Tensor.triu_

  

Tensor.true_divide

     

Tensor.true_divide_

     

Tensor.trunc

     

Tensor.trunc_

     

Tensor.type

     

Tensor.type_as

  

Tensor.unbind

  

Tensor.unflatten

  

Tensor.unfold

     

Tensor.uniform_

     

Tensor.unique

     

Tensor.unique_consecutive

     

Tensor.unsqueeze

  

Tensor.unsqueeze_

  

Tensor.values

     

Tensor.var

  

Tensor.vdot

  

Tensor.view

  

view

     

Tensor.view_as

  

Tensor.vsplit

     

Tensor.where

     

Tensor.xlogy

  

Tensor.xlogy_

     

Tensor.zero_