torch

Tensors

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

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	save	是

Parallelism

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

Locally disabling gradient computation

表6 Locally disabling gradient computation
序号	API名称	是否支持
1	enable_grad	是
2	inference_mode	否
3	is_grad_enabled	是
4	is_inference_mode_enabled	是
5	no_grad	是
6	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	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	否

Utilities

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

父主题： PyTorch1.11.0