functional.max_pool2d |
支持 |
|
functional.max_pool3d |
支持 |
|
functional.conv_transpose2d |
支持 |
|
functional.conv_transpose3d |
支持 |
|
functional.avg_pool2d |
Supported |
|
functional.avg_pool3d |
支持 |
|
functional.max_pool1d |
支持 |
|
functional.max_unpool1d |
支持 |
|
functional.max_unpool2d |
支持 |
|
functional.max_unpool3d |
支持 |
|
functional.lp_pool1d |
部分支持 |
功能存在限制 |
functional.lp_pool2d |
部分支持 |
功能存在限制 |
functional.adaptive_max_pool1d |
支持 |
|
functional.adaptive_max_pool2d |
支持 |
|
functional.adaptive_avg_pool1d |
支持 |
|
functional.fractional_max_pool2d |
支持 |
|
functional.fractional_max_pool3d |
支持 |
|
functional.threshold |
支持 |
|
functional.threshold_ |
支持 |
|
functional.relu |
支持 |
|
functional.relu_ |
支持 |
|
functional.hardtanh |
支持 |
|
functional.hardtanh_ |
支持 |
|
functional.hardswish |
支持 |
|
functional.relu6 |
支持 |
|
functional.elu |
支持 |
|
functional.elu_ |
支持 |
|
functional.selu |
支持 |
|
functional.celu |
支持 |
|
functional.leaky_relu |
支持 |
|
functional.leaky_relu_ |
支持 |
|
functional.prelu |
部分支持 |
inplace不支持图模式 |
functional.rrelu |
部分支持 |
inplace不支持图模式 |
functional.rrelu_ |
支持 |
|
functional.glu |
支持 |
|
functional.gelu |
支持 |
|
functional.logsigmoid |
支持 |
|
functional.hardshrink |
支持 |
|
functional.tanhshrink |
支持 |
|
functional.softsign |
支持 |
|
functional.softplus |
支持 |
|
functional.softmin |
支持 |
|
functional.softmax |
部分支持 |
功能存在限制 |
functional.softshrink |
支持 |
|
functional.gumbel_softmax |
支持 |
|
functional.log_softmax |
部分支持 |
功能存在限制 |
functional.tanh |
支持 |
|
functional.sigmoid |
支持 |
|
functional.hardsigmoid |
支持 |
|
functional.silu |
支持 |
|
functional.mish |
支持 |
|
functional.batch_norm |
支持 |
|
functional.group_norm |
支持 |
|
functional.instance_norm |
支持 |
|
functional.layer_norm |
支持 |
|
functional.local_response_norm |
支持 |
|
functional.normalize |
支持 |
|
functional.linear |
支持 |
|
functional.bilinear |
支持 |
|
functional.dropout |
支持 |
|
functional.alpha_dropout |
支持 |
|
functional.feature_alpha_dropout |
支持 |
|
functional.dropout1d |
部分支持 |
inplace不支持图模式 |
functional.dropout2d |
部分支持 |
inplace不支持图模式 |
functional.dropout3d |
部分支持 |
inplace不支持图模式 |
functional.one_hot |
支持 |
|
functional.pairwise_distance |
支持 |
|
functional.cosine_similarity |
支持 |
|
functional.pdist |
支持 |
|
functional.binary_cross_entropy |
支持 |
|
functional.binary_cross_entropy_with_logits |
支持 |
|
functional.poisson_nll_loss |
支持 |
|
functional.cosine_embedding_loss |
支持 |
|
functional.cross_entropy |
支持 |
|
functional.gaussian_nll_loss |
支持 |
|
functional.hinge_embedding_loss |
支持 |
|
functional.l1_loss |
支持 |
|
functional.mse_loss |
支持 |
|
functional.margin_ranking_loss |
支持 |
|
functional.multilabel_soft_margin_loss |
支持 |
|
functional.nll_loss |
支持 |
|
functional.smooth_l1_loss |
支持 |
|
functional.soft_margin_loss |
部分支持 |
暂不支持CPU后端 |
functional.triplet_margin_loss |
部分支持 |
只支持GPU后端 |
functional.triplet_margin_with_distance_loss |
支持 |
|
functional.pixel_shuffle |
支持 |
|
functional.pixel_unshuffle |
支持 |
|
functional.grid_sample |
支持 |
|
functional.huber_loss |
支持 |
|
functional.conv1d |
支持 |
|
functional.conv2d |
支持 |
|
functional.conv3d |
部分支持 |
功能存在限制 |
functional.conv_transpose1d |
支持 |
|
functional.unfold |
支持 |
|
functional.fold |
支持 |
|
functional.adaptive_max_pool3d |
部分支持 |
暂不支持Ascend后端 |
functional.adaptive_avg_pool2d |
支持 |
|
functional.adaptive_avg_pool3d |
支持 |
|
functional.embedding |
支持 |
|
functional.ctc_loss |
支持 |
|
functional.kl_div |
部分支持 |
功能存在限制 |
functional.multilabel_margin_loss |
部分支持 |
暂不支持CPU后端 |
functional.multi_margin_loss |
支持 |
|
functional.interpolate |
部分支持 |
功能存在限制 |
functional.upsample |
支持 |
|
functional.upsample_nearest |
支持 |
|
functional.upsample_bilinear |
部分支持 |
功能存在限制 |
functional.affine_grid |
支持 |
|
functional.avg_pool1d |
支持 |
|