torch

1

adjoint

否

2

arange

是

3

argwhere

否

4

as_strided

是

5

as_tensor

是

6

asarray

是

7

cat

是

8

chunk

是

9

column_stack

是

10

complex

否

11

concat

是

12

conj

否

13

dequantize

否

14

diagonal_scatter

是

15

dsplit

是

16

dstack

是

17

empty

是

18

empty_like

是

19

empty_strided

是

20

eye

是

21

from_numpy

是

22

frombuffer

否

23

full

是

24

full_like

是

25

gather

是

26

get_default_dtype

是，但是不支持NPU数据类型

27

heaviside

否

28

hsplit

是

29

hstack

是

30

index_add

是

31

index_select

是

32

is_complex

是，支持判断，但当前硬件限制不支持复数

33

is_conj

是

34

is_floating_point

是

35

is_nonzero

是

36

is_storage

是

37

is_tensor

是

38

linspace

是

39

logspace

是

40

masked_select

是

41

moveaxis

是

42

movedim

是

43

narrow

是

44

nonzero

是

45

numel

是

46

ones

是

47

ones_like

是

48

permute

是

49

polar

否

50

quantize_per_channel

是

51

quantize_per_tensor

是

52

range

是

53

reshape

是

54

row_stack

是

55

scatter

是

56

scatter_add

是

57

scatter_reduce

否

58

select

是

59

select_scatter

是

60

set_default_dtype

是，但是不支持NPU数据类型

61

set_default_tensor_type

是，但是不支持NPU数据类型

62

set_flush_denormal

是

63

set_printoptions

是

64

slice_scatter

是

65

sparse_coo_tensor

否

66

split

是

67

squeeze

是

68

stack

是

69

swapaxes

是

70

swapdims

是

71

t

是

72

take

是

73

take_along_dim

是

74

tensor

是

75

tensor_split

是

76

tile

是

77

transpose

是

78

unbind

是

79

unsqueeze

是

80

vsplit

是

81

vstack

是

82

where

是

83

zeros

是

84

zeros_like

是

1

Generator

是，但不支持get_state和set_state接口

1

bernoulli

是

2

get_rng_state

是

3

initial_seed

是

4

manual_seed

是

5

multinomial

是

6

normal

否

7

poisson

是

8

quasirandom.SobolEngine

是

9

rand

是

10

rand_like

是

11

randint

是

12

randint_like

是

13

randn

是

14

randn_like

是

15

randperm

是

16

seed

是

17

set_rng_state

是

18

torch.default_generator

是

19

torch.Tensor.bernoulli_

是

20

torch.Tensor.cauchy_

否

21

torch.Tensor.exponential_

否

22

torch.Tensor.geometric_

否

23

torch.Tensor.log_normal_

否

24

torch.Tensor.normal_

是

25

torch.Tensor.random_

是

26

torch.Tensor.uniform_

是

1

save

是

1

get_num_interop_threads

是

2

get_num_threads

是

3

load

是

4

set_num_interop_threads

是

5

set_num_threads

是

1

enable_grad

是

2

inference_mode

否

3

is_grad_enabled

是

4

is_inference_mode_enabled

是

5

no_grad

是

6

set_grad_enabled

是

1

abs

是

2

absolute

是

3

acos

是

4

acosh

是

5

add

是

6

addbmm

是

7

addcdiv

是

8

addcmul

是

9

addmm

是

10

addmv

是

11

addr

是

12

all

是

13

allclose

是

14

amax

是

15

amin

是

16

aminmax

否

17

angle

否

18

any

是

19

arccos

是

20

arccosh

是

21

arcsin

是

22

arcsinh

是

23

arctan

是

24

arctan2

是

25

arctanh

是

26

argmax

是

27

argmin

是

28

argsort

是

29

asin

是

30

asinh

是

31

atan

是

32

atan2

是

33

atanh

是

34

atleast_1d

是

35

atleast_2d

是

36

atleast_3d

是

37

baddbmm

是

38

bartlett_window

否

39

bincount

是

40

bitwise_and

是

41

bitwise_left_shift

否

42

bitwise_not

是

43

bitwise_or

是

44

bitwise_right_shift

否

45

bitwise_xor

是

46

blackman_window

否

47

block_diag

是

48

bmm

是

49

broadcast_shapes

是

50

broadcast_tensors

否

51

broadcast_to

是

52

bucketize

是

53

cartesian_prod

是

54

cdist

是，仅支持mode=donot_use_mm_for_euclid_dist

55

ceil

是

56

chain_matmul

是

57

cholesky

否

58

cholesky_inverse

否

59

cholesky_solve

否

60

clamp

是

61

clip

是

62

clone

是

63

combinations

是

64

conj_physical

否

65

copysign

是

66

corrcoef

是

67

cos

是

68

cosh

是

69

count_nonzero

否

70

cov

是

71

cross

是

72

cummax

是

73

cummin

是

74

cumprod

是

75

cumsum

是

76

cumulative_trapezoid

是

77

deg2rad

是

78

det

否

79

diag

是，仅支持diagonal=0场景

80

diag_embed

是

81

diagflat

是

82

diagonal

是

83

diff

是

84

digamma

是

85

dist

是

86

div

是

87

divide

是

88

dot

是

89

eig

否

90

einsum

是

91

eq

是

92

equal

是

93

erf

是

94

erfc

是

95

erfinv

是

96

exp

是

97

exp2

否

98

expm1

是

99

fake_quantize_per_channel_affine

否

100

fake_quantize_per_tensor_affine

否

101

fix

是

102

flatten

是

103

flip

是

104

fliplr

是

105

flipud

是

106

float_power

是

107

floor

是

108

floor_divide

是

109

fmax

否

110

fmin

否

111

fmod

是

112

frac

是

113

frexp

否

114

gcd

否

115

ge

是

116

geqrf

否

117

ger

是

118

gradient

是

119

greater

是

120

greater_equal

是

121

gt

是

122

hamming_window

否

123

hann_window

否

124

histc

否

125

histogram

否

126

histogramdd

否

127

hypot

否

128

i0

否

129

igamma

否

130

igammac

否

131

imag

否

132

inner

是

133

inverse

是

134

isclose

是

135

isfinite

是

136

isin

否

137

isinf

是

138

isnan

是

139

isneginf

是

140

isposinf

是

141

isreal

是

142

istft

否

143

kaiser_window

否

144

kron

是

145

kthvalue

否

146

lcm

否

147

ldexp

是

148

le

否

149

lerp

是

150

less

是

151

less_equal

是

152

lgamma

是

153

lobpcg

否

154

log

是

155

log10

是

156

log1p

是

157

log2

是

158

logaddexp

否

159

logaddexp2

否

160

logcumsumexp

否

161

logdet

否

162

logical_and

是

163

logical_not

是

164

logical_or

是

165

logical_xor

否

166

logit

是

167

logsumexp

是

168

lstsq

否

169

lt

否

170

lu

否

171

lu_solve

否

172

lu_unpack

否

173

matmul

是

174

matrix_exp

否

175

matrix_power

是

176

matrix_rank

是

177

max

是

178

maximum

是

179

mean

是

180

median

是

181

meshgrid

是

182

min

是

183

minimum

是

184

mm

是

185

mode

是

186

msort

是

187

mul

是

188

multiply

是

189

mv

是

190

mvlgamma

是

191

nan_to_num

否

192

nanmean

否

193

nanmedian

是

194

nanquantile

否

195

nansum

是

196

ne

否

197

neg

是

198

negative

是

199

nextafter

否

200

norm

是

201

not_equal

是

202

orgqr

否

203

ormqr

否

204

outer

是

205

pca_lowrank

是

206

pinverse

是

207

polygamma

否

208

positive

是

209

pow

是

210

prod

是

211

qr

是

212

quantile

是

213

quantized_batch_norm

否

214

quantized_max_pool1d

否

215

quantized_max_pool2d

否

216

rad2deg

是

217

ravel

是

218

real

是

219

reciprocal

是

220

remainder

是

221

renorm

是

222

repeat_interleave

是

223

resolve_conj

否

224

resolve_neg

否

225

roll

是

226

rot90

是

227

round

是

228

rsqrt

是

229

searchsorted

是

230

sgn

否

231

sigmoid

是

232

sign

是

233

signbit

否

234

sin

是

235

sinc

否

236

sinh

是

237

slogdet

是

238

solve

否

239

sort

否

240

sqrt

是

241

square

是

242

std

是

243

std_mean

是

244

stft

否

245

sub

是

246

subtract

是

247

sum

是

248

svd

是

249

svd_lowrank

是

250

symeig

是

251

tan

是

252

tanh

是

253

tensordot

是

254

topk

否

255

trace

否

256

trapezoid

是

257

trapz

是

258

triangular_solve

是

259

tril

是

260

tril_indices

是

261

triu

是

262

triu_indices

是

263

true_divide

是

264

trunc

是

265

unique

是

266

unique_consecutive

是，传参时必须使用关键字，否则精度不达标。return_inverse=return_inverse,return_counts=return_counts,dim=dim

267

vander

否

268

var

是

269

var_mean

是

270

vdot

是

271

view_as_complex

否

272

view_as_real

否

273

xlogy

否

1

_assert

是

2

are_deterministic_algorithms_enabled

是

3

can_cast

是

4

compiled_with_cxx11_abi

是

5

get_deterministic_debug_mode

是

6

is_deterministic_algorithms_warn_only_enabled

是

7

is_warn_always_enabled

是

8

promote_types

是

9

result_type

是

10

set_deterministic_debug_mode

是

11

set_warn_always

是

12

use_deterministic_algorithms

是