Tensors

表1 Tensors
序号	API名称	是否支持
1	arange	是
2	as_strided	是
3	as_tensor	是
4	cat	是
5	chunk	是
6	empty	是
7	empty_like	是
8	empty_strided	是
9	eye	是
10	from_numpy	是
11	full	是
12	full_like	是
13	gather	是
14	get_default_dtype	是
15	index_select	是
16	is_complex	是，支持判断，但当前硬件限制不支持复数
17	is_floating_point	是
18	is_storage	是
19	is_tensor	是
20	linspace	是
21	logspace	是
22	masked_select	是
23	narrow	是
24	nonzero	是
25	numel	是
26	ones	是
27	ones_like	是
28	quantize_per_channel	是
29	quantize_per_tensor	是
30	range	是
31	reshape	是
32	set_default_dtype	是
33	set_default_tensor_type	是
34	set_flush_denormal	是
35	set_printoptions	是
36	sparse_coo_tensor	否
37	split	是
38	squeeze	是
39	stack	是
40	t	是
41	take	是
42	tensor	是
43	transpose	是
44	unbind	是
45	unsqueeze	是
46	where	是
47	zeros	是
48	zeros_like	是

Generators

表2 Generators
序号	API名称	是否支持
1	Generator	是，但不支持get_state和set_state接口

Random sampling

表3 Random sampling
序号	API名称	是否支持
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_	是

Serialization

表4 Serialization
序号	API名称	是否支持
1	load	是
2	save	是

Parallelism

表5 Parallelism
序号	API名称	是否支持
1	get_num_interop_threads	是
2	get_num_threads	是
3	set_num_interop_threads	是
4	set_num_threads	是

Locally disabling gradient computation

表6 Locally disabling gradient computation
序号	API名称	是否支持
1	enable_grad	是
2	no_grad	是
3	set_grad_enabled	是

Math operations

表7 Math operations
序号	API名称	是否支持
1	abs	是
2	acos	是
3	add	是
4	addbmm	是
5	addcdiv	是
6	addcmul	是
7	addmm	是
8	addmv	是
9	addr	是
10	allclose	是
11	angle	否
12	argmax	是
13	argmin	是
14	argsort	是
15	asin	是
16	atan	是
17	atan2	是
18	baddbmm	是
19	bartlett_window	是
20	bincount	是
21	bitwise_and	是
22	bitwise_not	是
23	bitwise_or	是
24	bitwise_xor	是
25	blackman_window	是
26	bmm	是
27	broadcast_tensors	是
28	cartesian_prod	是
29	cdist	是，仅支持mode=donot_use_mm_for_euclid_dist
30	ceil	是
31	chain_matmul	是
32	cholesky	否
33	cholesky_inverse	否
34	cholesky_solve	否
35	clamp	是
36	combinations	否
37	conj	否
38	cos	是
39	cosh	是
40	cross	是
41	cummax	是
42	cummin	是
43	cumprod	是
44	cumsum	是
45	det	否
46	diag	是，仅支持diagonal=0场景
47	diag_embed	是
48	diagflat	是
49	diagonal	是
50	digamma	否
51	dist	是
52	div	是
53	dot	是
54	eig	否
55	einsum	是
56	eq	是
57	equal	是
58	erf	是
59	erfc	是
60	erfinv	是
61	exp	是
62	expm1	是
63	fft	否
64	flatten	是
65	flip	是
66	floor	是
67	floor_divide	是
68	fmod	是
69	frac	是
70	ge	是
71	geqrf	否
72	ger	是
73	gt	是
74	hamming_window	是
75	hann_window	是
76	histc	否
77	ifft	否
78	imag	否
79	inverse	是
80	isfinite	是
81	isinf	是
82	isnan	是
83	kthvalue	是
84	le	是
85	lerp	是
86	lgamma	否
87	lobpcg	否
88	log	是
89	log10	是
90	log1p	是
91	log2	是
92	logdet	否
93	logical_and	是
94	logical_not	是
95	logical_or	是
96	logical_xor	是
97	logsumexp	是
98	lstsq	否
99	lt	是
100	lu	否
101	lu_solve	否
102	lu_unpack	否
103	matmul	是
104	matrix_power	是
105	matrix_rank	是
106	max	是
107	mean	否
108	median	是
109	meshgrid	是
110	min	是
111	mm	是
112	mode	是
113	mul	是
114	mv	是
115	mvlgamma	否
116	ne	是
117	neg	是
118	norm	是
119	orgqr	否
120	ormqr	否
121	pca_lowrank	是
122	pinverse	是
123	polygamma	否
124	pow	是
125	prod	是
126	qr	是
127	real	是
128	reciprocal	是
129	remainder	是
130	renorm	是
131	repeat_interleave	是
132	rfft	否
133	rfft	否
134	roll	是
135	rot90	是
136	round	是
137	rsqrt	是
138	sigmoid	是
139	sign	是
140	sin	是
141	sinh	是
142	slogdet	是
143	solve	否
144	sort	是
145	sqrt	是
146	square	是
147	std	是
148	std_mean	是
149	stft	否
150	sum	是
151	svd	是
152	svd_lowrank	是
153	symeig	是
154	tan	是
155	tanh	是
156	tensordot	是
157	topk	是
158	trace	否
159	trapz	是
160	triangular_solve	是
161	tril	是
162	tril_indices	是
163	triu	是
164	triu_indices	是
165	true_divide	是
166	trunc	是
167	unique	否
168	unique_consecutive	否
169	var	是
170	var_mean	是

Utilities

表8 Utilities
序号	API名称	是否支持
1	can_cast	是
2	compiled_with_cxx11_abi	是
3	promote_types	是
4	result_type	是

torch

Tensors

Generators

Random sampling

Serialization

Parallelism

Locally disabling gradient computation

Math operations

Utilities