torch

Tensors

表1 Tensors
序号	API名称	是否支持
1	is_tensor	是
2	is_storage	是
3	is_complex	是，支持判断，但当前硬件限制不支持复数
4	is_conj	否
5	is_floating_point	是
6	is_nonzero	是
7	set_default_dtype	是，但是不支持NPU数据类型
8	get_default_dtype	是，但是不支持NPU数据类型
9	set_default_tensor_type	是，但是不支持NPU数据类型
10	numel	是
11	set_printoptions	是
12	set_flush_denormal	是
13	tensor	是
14	sparse_coo_tensor	否
15	asarray	是
16	as_tensor	是
17	as_strided	是
18	from_numpy	是
19	frombuffer	否
20	zeros	是
21	zeros_like	是
22	ones	是
23	ones_like	是
24	arange	是
25	range	是
26	linspace	是
27	logspace	是
28	eye	是
29	empty	是
30	empty_like	是
31	empty_strided	是
32	full	是
33	full_like	是
34	quantize_per_tensor	是
35	quantize_per_channel	是
36	dequantize	否
37	complex	否
38	polar	否
39	heaviside	否
40	adjoint	否
41	argwhere	否
42	cat	是
43	concat	是
44	conj	否
45	chunk	是
46	dsplit	是
47	column_stack	是
48	dstack	是
49	gather	是
50	hsplit	是
51	hstack	是
52	index_add	是
53	index_select	是
54	masked_select	是
55	movedim	是
56	moveaxis	是
57	narrow	是
58	nonzero	是
59	permute	是
60	reshape	是
61	row_stack	是
62	select	是
63	scatter	是
64	diagonal_scatter	是
65	select_scatter	是
66	slice_scatter	是
67	scatter_add	是
68	scatter_reduce	否
69	split	是
70	squeeze	是
71	stack	是
72	swapaxes	是
73	swapdims	是
74	t	是
75	take	是
76	take_along_dim	是
77	tensor_split	是
78	tile	是
79	transpose	是
80	unbind	是
81	unsqueeze	是
82	vsplit	是
83	vstack	是
84	where	是

Generators

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

Random sampling

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

Serialization

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

Parallelism

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

Locally disabling gradient computation

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

Math operations

表7 Math operations
序号	API名称	是否支持
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	arctan2	是
21	bitwise_not	是
22	bitwise_and	是
23	bitwise_or	是
24	bitwise_xor	是
25	bitwise_left_shift	否
26	bitwise_right_shift	否
27	ceil	是
28	clamp	是
29	clip	是
30	conj_physical	否
31	copysign	是
32	cos	是
33	cosh	是
34	deg2rad	是
35	div	是
36	divide	是
37	digamma	是
38	erf	是
39	erfc	是
40	erfinv	是
41	exp	是
42	exp2	否
43	expm1	是
44	fake_quantize_per_channel_affine	否
45	fake_quantize_per_tensor_affine	否
46	fix	是
47	float_power	是
48	floor	是
49	floor_divide	是
50	fmod	是
51	frac	是
52	frexp	否
53	gradient	是
54	imag	否
55	ldexp	是
56	lerp	是
57	lgamma	是
58	log	是
59	log10	是
60	log1p	是
61	log2	是
62	logaddexp	否
63	logaddexp2	否
64	logical_and	是
65	logical_not	是
66	logical_or	是
67	logical_xor	否
68	logit	是
69	hypot	否
70	i0	否
71	igamma	否
72	igammac	否
73	mul	是
74	multiply	是
75	mvlgamma	是
76	nan_to_num	否
77	neg	是
78	negative	是
79	nextafter	否
80	polygamma	否
81	positive	是
82	pow	是
83	quantized_batch_norm	否
84	quantized_max_pool1d	否
85	quantized_max_pool2d	否
86	rad2deg	是
87	real	是
88	reciprocal	是
89	remainder	是
90	round	是
91	rsqrt	是
92	sigmoid	是
93	sign	是
94	sgn	否
95	signbit	否
96	sin	是
97	sinc	否
98	sinh	是
99	sqrt	是
100	square	是
101	sub	是
102	subtract	是
103	tan	是
104	tanh	是
105	true_divide	是
106	trunc	是
107	xlogy	否
108	argmax	是
109	argmin	是
110	amax	是
111	amin	是
112	aminmax	否
113	all	是
114	any	是
115	max	是
116	min	是
117	dist	是
118	logsumexp	是
119	mean	是
120	nanmean	否
121	median	是
122	nanmedian	是
123	mode	是
124	norm	是
125	nansum	是
126	prod	是
127	quantile	是
128	nanquantile	否
129	std	是
130	std_mean	是
131	sum	是
132	unique	是
133	unique_consecutive	是，传参时必须使用关键字，否则精度不达标。return_inverse=return_inverse,return_counts=return_counts,dim=dim
134	var	是
135	var_mean	是
136	count_nonzero	否
137	allclose	是
138	argsort	是
139	eq	是
140	equal	是
141	ge	是
142	greater_equal	是
143	gt	是
144	greater	是
145	isclose	是
146	isfinite	是
147	isin	否
148	isinf	是
149	isposinf	是
150	isneginf	是
151	isnan	是
152	isreal	是
153	kthvalue	否
154	le	否
155	less_equal	是
156	lt	否
157	less	是
158	maximum	是
159	minimum	是
160	fmax	否
161	fmin	否
162	ne	否
163	not_equal	是
164	sort	否
165	topk	否
166	msort	是
167	stft	否
168	istft	否
169	bartlett_window	否
170	blackman_window	否
171	hamming_window	否
172	hann_window	否
173	kaiser_window	否
174	atleast_1d	是
175	atleast_2d	是
176	atleast_3d	是
177	bincount	是
178	block_diag	是
179	broadcast_tensors	否
180	broadcast_to	是
181	broadcast_shapes	是
182	bucketize	是
183	cartesian_prod	是
184	cdist	是，仅支持mode=donot_use_mm_for_euclid_dist
185	clone	是
186	combinations	是
187	corrcoef	是
188	cov	是
189	cross	是
190	cummax	是
191	cummin	是
192	cumprod	是
193	cumsum	是
194	diag	是，仅支持diagonal=0场景
195	diag_embed	是
196	diagflat	是
197	diagonal	是
198	diff	是
199	einsum	是
200	flatten	是
201	flip	是
202	fliplr	是
203	flipud	是
204	kron	是
205	rot90	是
206	gcd	否
207	histc	否
208	histogram	否
209	histogramdd	否
210	meshgrid	是
211	lcm	否
212	logcumsumexp	否
213	ravel	是
214	renorm	是
215	repeat_interleave	是
216	roll	是
217	searchsorted	是
218	tensordot	是
219	trace	否
220	tril	是
221	tril_indices	是
222	triu	是
223	triu_indices	是
224	vander	否
225	view_as_real	否
226	view_as_complex	否
227	resolve_conj	否
228	resolve_neg	否
229	addbmm	是
230	addmm	是
231	addmv	是
232	addr	是
233	baddbmm	是
234	bmm	是
235	chain_matmul	是
236	cholesky	否
237	cholesky_inverse	否
238	cholesky_solve	否
239	dot	是
240	eig	否
241	geqrf	否
242	ger	是
243	inner	是
244	inverse	是
245	det	否
246	logdet	否
247	slogdet	是
248	lstsq	否
249	lu	否
250	lu_solve	否
251	lu_unpack	否
252	matmul	是
253	matrix_power	是
254	matrix_rank	是
255	matrix_exp	否
256	mm	是
257	mv	是
258	orgqr	否
259	ormqr	否
260	outer	是
261	pinverse	是
262	qr	是
263	solve	否
264	svd	是
265	svd_lowrank	是
266	pca_lowrank	是
267	symeig	是
268	lobpcg	否
269	trapz	是
270	trapezoid	是
271	cumulative_trapezoid	是
272	triangular_solve	是
273	vdot	是

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	is_deterministic_algorithms_warn_only_enabled	是
8	set_deterministic_debug_mode	是
9	get_deterministic_debug_mode	是
10	set_warn_always	是
11	is_warn_always_enabled	是
12	_assert	是

父主题： PyTorch1.11.0