torch

Tensors

表1 Tensors
序号	API名称	是否支持
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	是

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	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	否

Utilities

表8 Utilities
序号	API名称	是否支持
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	是

父主题： PyTorch1.8.1