torch

1

is_tensor

是

2

is_storage

是

3

is_complex

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

4

is_floating_point

是

5

is_nonzero

是

6

set_default_dtype

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

7

get_default_dtype

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

8

set_default_tensor_type

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

9

numel

是

10

set_printoptions

是

11

set_flush_denormal

是

12

tensor

是

13

sparse_coo_tensor

否

14

as_tensor

是

15

as_strided

是

16

from_numpy

是

17

zeros

是

18

zeros_like

是

19

ones

是

20

ones_like

是

21

arange

是

22

range

是

23

linspace

是

24

logspace

是

25

eye

是

26

empty

是

27

empty_like

是

28

empty_strided

是

29

full

是

30

full_like

是

31

quantize_per_tensor

是

32

quantize_per_channel

是

33

dequantize

否

34

complex

否

35

polar

否

36

heaviside

否

37

cat

是

38

chunk

是

39

column_stack

是

40

dstack

是

41

gather

是

42

hstack

是

43

index_select

是

44

masked_select

是

45

movedim

是

46

moveaxis

是

47

narrow

是

48

nonzero

是

49

reshape

是

50

row_stack

是

51

scatter

是

52

scatter_add

是

53

split

是

54

squeeze

是

55

stack

是

56

swapaxes

是

57

swapdims

是

58

t

是

59

take

是

60

tensor_split

是

61

tile

是

62

transpose

是

63

unbind

是

64

unsqueeze

是

65

vstack

是

66

where

是

1

Generator

是，但不支持get_state和set_state接口

1

seed

是

2

manual_seed

是

3

initial_seed

是

4

get_rng_state

是

5

set_rng_state

是

6

torch.default_generator

是

7

bernoulli

是

8

multinomial

是

9

normal

否

10

poisson

是

11

rand

是

12

rand_like

是

13

randint

是

14

randint_like

是

15

randn

是

16

randn_like

是

17

randperm

是

18

torch.Tensor.bernoulli_

是

19

torch.Tensor.cauchy_

否

20

torch.Tensor.exponential_

否

21

torch.Tensor.geometric_

否

22

torch.Tensor.log_normal_

否

23

torch.Tensor.normal_

是

24

torch.Tensor.random_

是

25

torch.Tensor.uniform_

是

26

quasirandom.SobolEngine

是

1

save

是

2

load

是

1

get_num_threads

是

2

set_num_threads

是

3

get_num_interop_threads

是

4

set_num_interop_threads

是

1

no_grad

是

2

enable_grad

是

3

set_grad_enabled

是

1

abs

是

2

absolute

是

3

acos

是

4

arccos

是

5

acosh

是

6

arccosh

是

7

add

是

8

addcdiv

是

9

addcmul

是

10

angle

否

11

asin

是

12

arcsin

是

13

asinh

是

14

arcsinh

是

15

atan

是

16

arctan

是

17

atanh

是

18

arctanh

是

19

atan2

是

20

bitwise_not

是

21

bitwise_and

是

22

bitwise_or

是

23

bitwise_xor

是

24

ceil

是

25

clamp

是

26

clip

是

27

conj

否

28

copysign

是

29

cos

是

30

cosh

是

31

deg2rad

是

32

div

是

33

divide

是

34

digamma

是

35

erf

是

36

erfc

是

37

erfinv

是

38

exp

是

39

exp2

否

40

expm1

是

41

fake_quantize_per_channel_affine

否

42

fake_quantize_per_tensor_affine

否

43

fix

是

44

float_power

是

45

floor

是

46

floor_divide

是

47

fmod

是

48

frac

是

49

imag

否

50

ldexp

是

51

lerp

是

52

lgamma

是

53

log

是

54

log10

是

55

log1p

是

56

log2

是

57

logaddexp

否

58

logaddexp2

否

59

logical_and

是

60

logical_not

是

61

logical_or

是

62

logical_xor

否

63

logit

是

64

hypot

否

65

i0

否

66

igamma

否

67

igammac

否

68

mul

是

69

multiply

是

70

mvlgamma

是

71

nan_to_num

否

72

neg

是

73

negative

是

74

nextafter

否

75

polygamma

否

76

pow

是

77

rad2deg

是

78

real

是

79

reciprocal

是

80

remainder

是

81

round

是

82

rsqrt

是

83

sigmoid

是

84

sign

是

85

sgn

否

86

signbit

否

87

sin

是

88

sinc

否

89

sinh

是

90

sqrt

是

91

square

是

92

sub

是

93

subtract

是

94

tan

是

95

tanh

是

96

true_divide

是

97

trunc

是

98

xlogy

否

99

argmax

是

100

argmin

是

101

amax

是

102

amin

是

103

all

是

104

any

是

105

max

是

106

min

是

107

dist

是

108

logsumexp

是

109

mean

是

110

median

是

111

nanmedian

是

112

mode

是

113

norm

是

114

nansum

是

115

prod

是

116

quantile

是

117

nanquantile

否

118

std

是

119

std_mean

是

120

sum

是

121

unique

是

122

unique_consecutive

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

123

var

是

124

var_mean

是

125

count_nonzero

否

126

allclose

是

127

argsort

是

128

eq

是

129

equal

是

130

ge

是

131

greater_equal

是

132

gt

是

133

greater

是

134

isclose

是

135

isfinite

是

136

isinf

是

137

isposinf

是

138

isneginf

是

139

isnan

是

140

isreal

是

141

kthvalue

否

142

le

否

143

less_equal

是

144

lt

否

145

less

是

146

maximum

是

147

minimum

是

148

fmax

否

149

fmin

否

150

ne

否

151

not_equal

是

152

sort

否

153

topk

否

154

msort

是

155

stft

否

156

istft

否

157

bartlett_window

否

158

blackman_window

否

159

hamming_window

否

160

hann_window

否

161

kaiser_window

否

162

atleast_1d

是

163

atleast_2d

是

164

atleast_3d

是

165

bincount

是

166

block_diag

是

167

broadcast_tensors

否

168

broadcast_to

是

169

broadcast_shapes

是

170

bucketize

是

171

cartesian_prod

是

172

cdist

是，仅支持mode=donot_use_mm_for_euclid_dist

173

clone

是

174

combinations

是

175

cross

是

176

cummax

是

177

cummin

是

178

cumprod

是

179

cumsum

是

180

diag

是，仅支持diagonal=0场景

181

diag_embed

是

182

diagflat

是

183

diagonal

是

184

diff

是

185

einsum

是

186

flatten

是

187

flip

是

188

fliplr

是

189

flipud

是

190

kron

是

191

rot90

是

192

gcd

否

193

histc

否

194

meshgrid

是

195

lcm

否

196

logcumsumexp

否

197

ravel

是

198

renorm

是

199

repeat_interleave

是

200

roll

是

201

searchsorted

是

202

tensordot

是

203

trace

否

204

tril

是

205

tril_indices

是

206

triu

是

207

triu_indices

是

208

vander

否

209

view_as_real

否

210

view_as_complex

否

211

addbmm

是

212

addmm

是

213

addmv

是

214

addr

是

215

baddbmm

是

216

bmm

是

217

chain_matmul

是

218

cholesky

否

219

cholesky_inverse

否

220

cholesky_solve

否

221

dot

是

222

eig

否

223

geqrf

否

224

ger

是

225

inner

是

226

inverse

是

227

det

否

228

logdet

否

229

slogdet

是

230

lstsq

否

231

lu

否

232

lu_solve

否

233

lu_unpack

否

234

matmul

是

235

matrix_power

是

236

matrix_rank

是

237

matrix_exp

否

238

mm

是

239

mv

是

240

orgqr

否

241

ormqr

否

242

outer

是

243

pinverse

是

244

qr

是

245

solve

否

246

svd

是

247

svd_lowrank

是

248

pca_lowrank

是

249

symeig

是

250

lobpcg

否

251

trapz

是

252

triangular_solve

是

253

vdot

是

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

是