序号 |
API名称 |
是否支持 |
|---|---|---|
1 |
conv_transpose1d |
是 |
2 |
conv_transpose2d |
是 |
3 |
conv_transpose3d |
是 |
4 |
conv1d |
是 |
5 |
conv2d |
是 |
6 |
conv3d |
是 |
7 |
fold |
是 |
8 |
unfold |
是 |
序号 |
API名称 |
是否支持 |
|---|---|---|
1 |
adaptive_avg_pool1d |
是 |
2 |
adaptive_avg_pool2d |
是 |
3 |
adaptive_avg_pool3d |
是，仅支持D=1，H=1，W=1场景 |
4 |
adaptive_max_pool1d |
是 |
5 |
adaptive_max_pool2d |
是 |
6 |
adaptive_max_pool3d |
否 |
7 |
avg_pool1d |
是 |
8 |
avg_pool2d |
是 |
9 |
avg_pool3d |
是 |
10 |
lp_pool1d |
是 |
11 |
lp_pool2d |
是 |
12 |
max_pool1d |
是 |
13 |
max_pool2d |
是 |
14 |
max_pool3d |
是 |
15 |
max_unpool1d |
是 |
16 |
max_unpool2d |
是 |
17 |
max_unpool3d |
是 |
序号 |
API名称 |
是否支持 |
|---|---|---|
1 |
celu |
是 |
2 |
elu |
是 |
3 |
elu_ |
是 |
4 |
gelu |
是 |
5 |
glu |
是 |
6 |
gumbel_softmax |
否 |
7 |
hardshrink |
是 |
8 |
hardsigmoid |
是 |
9 |
hardswish |
是 |
10 |
hardtanh |
是 |
11 |
hardtanh_ |
是 |
12 |
leaky_relu |
是 |
13 |
leaky_relu_ |
是 |
14 |
log_softmax |
是 |
15 |
logsigmoid |
是 |
16 |
prelu |
是 |
17 |
relu |
是 |
18 |
relu_ |
是 |
19 |
relu6 |
是 |
20 |
rrelu |
是 |
21 |
rrelu_ |
是 |
22 |
selu |
是 |
23 |
sigmoid |
是 |
24 |
silu |
是 |
25 |
softmax |
是 |
26 |
softmin |
是 |
27 |
softplus |
是 |
28 |
softshrink |
是 |
29 |
softsign |
是 |
30 |
tanh |
是 |
31 |
tanhshrink |
是 |
32 |
threshold |
是 |
33 |
threshold_ |
是 |
序号 |
API名称 |
是否支持 |
|---|---|---|
1 |
batch_norm |
是 |
2 |
instance_norm |
是 |
3 |
layer_norm |
是 |
4 |
local_response_norm |
是 |
5 |
normalize |
是 |
序号 |
API名称 |
是否支持 |
|---|---|---|
1 |
bilinear |
是。Atlas A2训练系列产品上的输入不支持int32 |
2 |
linear |
是，不支持int32 |
序号 |
API名称 |
是否支持 |
|---|---|---|
1 |
alpha_dropout |
是 |
2 |
dropout |
是 |
3 |
dropout2d |
是 |
4 |
dropout3d |
是 |
5 |
feature_alpha_dropout |
是 |
序号 |
API名称 |
是否支持 |
|---|---|---|
1 |
embedding |
是 |
2 |
embedding_bag |
是 |
3 |
one_hot |
是 |
序号 |
API名称 |
是否支持 |
|---|---|---|
1 |
cosine_similarity |
是 |
2 |
pairwise_distance |
是 |
3 |
pdist |
是 |
序号 |
API名称 |
是否支持 |
|---|---|---|
1 |
binary_cross_entropy |
是 |
2 |
binary_cross_entropy_with_logits |
是 |
3 |
cosine_embedding_loss |
是 |
4 |
cross_entropy |
是 |
5 |
ctc_loss |
是 |
6 |
hinge_embedding_loss |
是，只支持fp32 |
7 |
kl_div |
是 |
8 |
l1_loss |
是 |
9 |
margin_ranking_loss |
是 |
10 |
mse_loss |
是 |
11 |
multi_margin_loss |
否，可以走CPU实现 |
12 |
multilabel_margin_loss |
是 |
13 |
multilabel_soft_margin_loss |
是，仅支持fp32,fp16，输入仅支持第一个输入2D,第二个输入1D |
14 |
nll_loss |
是 |
15 |
poisson_nll_loss |
是 |
16 |
smooth_l1_loss |
是 |
17 |
soft_margin_loss |
是 |
18 |
triplet_margin_loss |
是 |
19 |
triplet_margin_with_distance_loss |
是 |
序号 |
API名称 |
是否支持 |
|---|---|---|
1 |
affine_grid |
是 |
2 |
grid_sample |
是 |
3 |
interpolate |
是，不支持recompute_scale_factor=True的场景 |
4 |
pad |
是 |
5 |
pixel_shuffle |
是，不支持8维输入 |
6 |
pixel_unshuffle |
是，不支持8维输入 |
7 |
upsample |
Atlas 训练系列产品：是 Atlas A2训练系列产品：否 |
8 |
upsample_bilinear |
是 |
9 |
upsample_nearest |
是 |
序号 |
API名称 |
是否支持 |
|---|---|---|
1 |
torch.nn.parallel.data_parallel |
否 |