torch

1

arange

是

2

as_strided

是

3

as_tensor

是

4

cat

是

5

chunk

是

6

column_stack

是

7

complex

否

8

dequantize

否

9

dstack

是

10

empty

是

11

empty_like

是

12

empty_strided

是

13

eye

是

14

from_numpy

是

15

full

是

16

full_like

是

17

gather

是

18

get_default_dtype

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

19

heaviside

否

20

hstack

是

21

index_select

是

22

is_complex

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

23

is_floating_point

是

24

is_nonzero

是

25

is_storage

是

26

is_tensor

是

27

linspace

是

28

logspace

是

29

masked_select

是

30

moveaxis

是

31

movedim

是

32

narrow

是

33

nonzero

是

34

numel

是

35

ones

是

36

ones_like

是

37

polar

否

38

quantize_per_channel

是

39

quantize_per_tensor

是

40

range

是

41

reshape

是

42

row_stack

是

43

scatter

是

44

scatter_add

是

45

set_default_dtype

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

46

set_default_tensor_type

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

47

set_flush_denormal

是

48

set_printoptions

是

49

sparse_coo_tensor

否

50

split

是

51

squeeze

是

52

stack

是

53

swapaxes

是

54

swapdims

是

55

t

是

56

take

是

57

tensor

是

58

tensor_split

是

59

tile

是

60

transpose

是

61

unbind

是

62

unsqueeze

是

63

vstack

是

64

where

是

65

zeros

是

66

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

load

是

2

save

是

1

get_num_interop_threads

是

2

get_num_threads

是

3

set_num_interop_threads

是

4

set_num_threads

是

1

enable_grad

是

2

no_grad

是

3

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

angle

否

17

any

是

18

arccos

是

19

arccosh

是

20

arcsin

是

21

arcsinh

是

22

arctan

是

23

arctanh

是

24

argmax

是

25

argmin

是

26

argsort

是

27

asin

是

28

asinh

是

29

atan

是

30

atan2

是

31

atanh

是

32

atleast_1d

是

33

atleast_2d

是

34

atleast_3d

是

35

baddbmm

是

36

bartlett_window

否

37

bincount

是

38

bitwise_and

是

39

bitwise_not

是

40

bitwise_or

是

41

bitwise_xor

是

42

blackman_window

否

43

block_diag

是

44

bmm

是

45

broadcast_shapes

是

46

broadcast_tensors

否

47

broadcast_to

是

48

bucketize

是

49

cartesian_prod

是

50

cdist

是，仅支持mode=donot_use_mm_for_euclid_dist

51

ceil

是

52

chain_matmul

是

53

cholesky

否

54

cholesky_inverse

否

55

cholesky_solve

否

56

clamp

是

57

clip

是

58

clone

是

59

combinations

是

60

conj

否

61

copysign

是

62

cos

是

63

cosh

是

64

count_nonzero

否

65

cross

是

66

cummax

是

67

cummin

是

68

cumprod

是

69

cumsum

是

70

deg2rad

是

71

det

否

72

diag

是，仅支持diagonal=0场景

73

diag_embed

是

74

diagflat

是

75

diagonal

是

76

diff

是

77

digamma

是

78

dist

是

79

div

是

80

divide

是

81

dot

是

82

eig

否

83

einsum

是

84

eq

是

85

equal

是

86

erf

是

87

erfc

是

88

erfinv

是

89

exp

是

90

exp2

否

91

expm1

是

92

fake_quantize_per_channel_affine

否

93

fake_quantize_per_tensor_affine

否

94

fix

是

95

flatten

是

96

flip

是

97

fliplr

是

98

flipud

是

99

float_power

是

100

floor

是

101

floor_divide

是

102

fmax

否

103

fmin

否

104

fmod

是

105

frac

是

106

gcd

否

107

ge

是

108

geqrf

否

109

ger

是

110

greater

是

111

greater_equal

是

112

gt

是

113

hamming_window

否

114

hann_window

否

115

histc

否

116

hypot

否

117

i0

否

118

igamma

否

119

igammac

否

120

imag

否

121

inner

是

122

inverse

是

123

isclose

是

124

isfinite

是

125

isinf

是

126

isnan

是

127

isneginf

是

128

isposinf

是

129

isreal

是

130

istft

否

131

kaiser_window

否

132

kron

是

133

kthvalue

否

134

lcm

否

135

ldexp

是

136

le

否

137

lerp

是

138

less

是

139

less_equal

是

140

lgamma

是

141

lobpcg

否

142

log

是

143

log10

是

144

log1p

是

145

log2

是

146

logaddexp

否

147

logaddexp2

否

148

logcumsumexp

否

149

logdet

否

150

logical_and

是

151

logical_not

是

152

logical_or

是

153

logical_xor

否

154

logit

是

155

logsumexp

是

156

lstsq

否

157

lt

否

158

lu

否

159

lu_solve

否

160

lu_unpack

否

161

matmul

是

162

matrix_exp

否

163

matrix_power

是

164

matrix_rank

是

165

max

是

166

maximum

是

167

mean

是

168

median

是

169

meshgrid

是

170

min

是

171

minimum

是

172

mm

是

173

mode

是

174

msort

是

175

mul

是

176

multiply

是

177

mv

是

178

mvlgamma

是

179

nan_to_num

否

180

nanmedian

是

181

nanquantile

否

182

nansum

是

183

ne

否

184

neg

是

185

negative

是

186

nextafter

否

187

norm

是

188

not_equal

是

189

orgqr

否

190

ormqr

否

191

outer

是

192

pca_lowrank

是

193

pinverse

是

194

polygamma

否

195

pow

是

196

prod

是

197

qr

是

198

quantile

是

199

rad2deg

是

200

ravel

是

201

real

是

202

reciprocal

是

203

remainder

是

204

renorm

是

205

repeat_interleave

是

206

roll

是

207

rot90

是

208

round

是

209

rsqrt

是

210

searchsorted

是

211

sgn

否

212

sigmoid

是

213

sign

是

214

signbit

否

215

sin

是

216

sinc

否

217

sinh

是

218

slogdet

是

219

solve

否

220

sort

否

221

sqrt

是

222

square

是

223

std

是

224

std_mean

是

225

stft

否

226

sub

是

227

subtract

是

228

sum

是

229

svd

是

230

svd_lowrank

是

231

symeig

是

232

tan

是

233

tanh

是

234

tensordot

是

235

topk

否

236

trace

否

237

trapz

是

238

triangular_solve

是

239

tril

是

240

tril_indices

是

241

triu

是

242

triu_indices

是

243

true_divide

是

244

trunc

是

245

unique

是

246

unique_consecutive

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

247

vander

否

248

var

是

249

var_mean

是

250

vdot

是

251

view_as_complex

否

252

view_as_real

否

253

xlogy

否

1

compiled_with_cxx11_abi

是

2

result_type

是

3

can_cast

是

4

promote_types

是

5

use_deterministic_algorithms

是

6

are_deterministic_algorithms_enabled

是

7

_assert

是