序号 |
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 |
hardtanh |
是 |
9 |
hardtanh_ |
是 |
10 |
leaky_relu |
是 |
11 |
leaky_relu_ |
是 |
12 |
log_softmax |
是 |
13 |
logsigmoid |
是 |
14 |
prelu |
是 |
15 |
relu |
是 |
16 |
relu_ |
是 |
17 |
relu6 |
是 |
18 |
rrelu |
是 |
19 |
rrelu_ |
是 |
20 |
selu |
是 |
21 |
sigmoid |
是 |
22 |
softmax |
是 |
23 |
softmin |
是 |
24 |
softplus |
是 |
25 |
softshrink |
是 |
26 |
softsign |
是 |
27 |
tanh |
是 |
28 |
tanhshrink |
是 |
29 |
threshold |
是 |
30 |
threshold_ |
是 |
序号 |
API名称 |
是否支持 |
|---|---|---|
1 |
batch_norm |
是 |
2 |
instance_norm |
是 |
3 |
layer_norm |
是 |
4 |
local_response_norm |
是 |
5 |
normalize |
是 |
序号 |
API名称 |
是否支持 |
|---|---|---|
1 |
bilinear |
是 |
2 |
linear |
是 |
序号 |
API名称 |
是否支持 |
|---|---|---|
1 |
alpha_dropout |
是 |
2 |
dropout |
是 |
3 |
dropout2d |
是 |
4 |
dropout3d |
是 |
序号 |
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 |
是，仅支持2维输入 |
6 |
hinge_embedding_loss |
是 |
7 |
kl_div |
是 |
8 |
l1_loss |
是 |
9 |
margin_ranking_loss |
是 |
10 |
mse_loss |
是 |
11 |
multi_margin_loss |
否 |
12 |
multilabel_margin_loss |
是 |
13 |
multilabel_soft_margin_loss |
是 |
14 |
nll_loss |
是 |
15 |
poisson_nll_loss |
是 |
16 |
smooth_l1_loss |
是 |
17 |
soft_margin_loss |
是 |
18 |
triplet_margin_loss |
是 |
序号 |
API名称 |
是否支持 |
|---|---|---|
1 |
affine_grid |
是 |
2 |
grid_sample |
是 |
3 |
interpolate |
是 |
4 |
pad |
是 |
5 |
pixel_shuffle |
是 |
6 |
upsample |
是 |
7 |
upsample_bilinear |
是 |
8 |
upsample_nearest |
是 |
序号 |
API名称 |
是否支持 |
|---|---|---|
1 |
torch.nn.parallel.data_parallel |
否 |