序号 |
API名称 |
是否支持 |
|---|---|---|
1 |
conv1d |
是 |
2 |
conv2d |
是 |
3 |
conv3d |
是 |
4 |
conv_transpose1d |
是 |
5 |
conv_transpose2d |
是 |
6 |
conv_transpose3d |
是 |
7 |
unfold |
是 |
8 |
fold |
是 |
序号 |
API名称 |
是否支持 |
|---|---|---|
1 |
avg_pool1d |
是 |
2 |
avg_pool2d |
是 |
3 |
avg_pool3d |
是 |
4 |
max_pool1d |
是 |
5 |
max_pool2d |
是 |
6 |
max_pool3d |
是 |
7 |
max_unpool1d |
是 |
8 |
max_unpool2d |
是 |
9 |
max_unpool3d |
是 |
10 |
lp_pool1d |
是 |
11 |
lp_pool2d |
是 |
12 |
adaptive_max_pool1d |
是 |
13 |
adaptive_max_pool2d |
是 |
14 |
adaptive_max_pool3d |
否 |
15 |
adaptive_avg_pool1d |
是 |
16 |
adaptive_avg_pool2d |
是 |
17 |
adaptive_avg_pool3d |
是，仅支持D=1，H=1，W=1场景 |
18 |
fractional_max_pool2d |
否 |
19 |
fractional_max_pool3d |
否 |
序号 |
API名称 |
是否支持 |
|---|---|---|
1 |
threshold |
是 |
2 |
threshold_ |
是 |
3 |
relu |
是 |
4 |
relu_ |
是 |
5 |
hardtanh |
是 |
6 |
hardtanh_ |
是 |
7 |
hardswish |
是 |
8 |
relu6 |
是 |
9 |
elu |
是 |
10 |
elu_ |
是 |
11 |
selu |
是 |
12 |
celu |
是 |
13 |
leaky_relu |
是 |
14 |
leaky_relu_ |
是 |
15 |
prelu |
是 |
16 |
rrelu |
是 |
17 |
rrelu_ |
是 |
18 |
glu |
是 |
19 |
gelu |
是 |
20 |
logsigmoid |
是 |
21 |
hardshrink |
是 |
22 |
tanhshrink |
是 |
23 |
softsign |
是 |
24 |
softplus |
是 |
25 |
softmin |
是 |
26 |
softmax |
是 |
27 |
softshrink |
是 |
28 |
gumbel_softmax |
否 |
29 |
log_softmax |
是 |
30 |
tanh |
是 |
31 |
sigmoid |
是 |
32 |
hardsigmoid |
是 |
33 |
silu |
是 |
34 |
mish |
否 |
35 |
batch_norm |
是 |
36 |
group_norm |
否 |
37 |
instance_norm |
是 |
38 |
layer_norm |
是 |
39 |
local_response_norm |
是 |
40 |
normalize |
是 |
序号 |
API名称 |
是否支持 |
|---|---|---|
1 |
linear |
是 |
2 |
bilinear |
是 |
序号 |
API名称 |
是否支持 |
|---|---|---|
1 |
dropout |
是 |
2 |
alpha_dropout |
是 |
3 |
feature_alpha_dropout |
是 |
4 |
dropout2d |
是 |
5 |
dropout3d |
是 |
序号 |
API名称 |
是否支持 |
|---|---|---|
1 |
embedding |
是 |
2 |
embedding_bag |
是 |
3 |
one_hot |
是 |
序号 |
API名称 |
是否支持 |
|---|---|---|
1 |
pairwise_distance |
是 |
2 |
cosine_similarity |
是 |
3 |
pdist |
是 |
序号 |
API名称 |
是否支持 |
|---|---|---|
1 |
binary_cross_entropy |
是，参数y仅支持y=1,y=0 |
2 |
binary_cross_entropy_with_logits |
是 |
3 |
poisson_nll_loss |
是 |
4 |
cosine_embedding_loss |
是 |
5 |
cross_entropy |
是 |
6 |
ctc_loss |
是，仅支持2维输入 |
7 |
gaussian_nll_loss |
是 |
8 |
hinge_embedding_loss |
是 |
9 |
kl_div |
是 |
10 |
l1_loss |
是 |
11 |
mse_loss |
是 |
12 |
margin_ranking_loss |
是 |
13 |
multilabel_margin_loss |
是 |
14 |
multilabel_soft_margin_loss |
是 |
15 |
multi_margin_loss |
是 |
16 |
nll_loss |
是 |
17 |
huber_loss |
否 |
18 |
smooth_l1_loss |
是 |
19 |
soft_margin_loss |
是 |
20 |
triplet_margin_loss |
是 |
21 |
triplet_margin_with_distance_loss |
是 |
序号 |
API名称 |
是否支持 |
|---|---|---|
1 |
pixel_shuffle |
是 |
2 |
pixel_unshuffle |
是 |
3 |
pad |
是 |
4 |
interpolate |
是 |
5 |
upsample |
是 |
6 |
upsample_nearest |
是 |
7 |
upsample_bilinear |
是 |
8 |
grid_sample |
是 |
9 |
affine_grid |
是 |
序号 |
API名称 |
是否支持 |
|---|---|---|
1 |
torch.nn.parallel.data_parallel |
否 |