supported_API.md

# PyTorch训练项目转换支持API列表
> 目前PyTorch训练项目转换支持6个优化器相关API，70+的NN类API，10+Utils类API，2个Autograd类API，40+的基础操作API以及30+Torchvision API，我们在如下列表中给出了目前的全部API。

## 优化器相关API
| 序号 | API                                        | 序号 | API                                        |
| ---- | ------------------------------------------ | ---- | ------------------------------------------ |
| 1    | torch.optim                                | 2    | torch.optim.lr_scheduler.ReduceLROnPlateau |
| 3    | torch.optim.lr_scheduler.CosineAnnealingLR | 4    | torch.optim.lr_scheduler.MultiStepLR       |
| 5    | torch.optim.Adam                           | 6    | torch.optim.SGD                            |

## NN类API

| 序号 | API                                                  | 序号 | API                               |
| ---- | ---------------------------------------------------- | ---- | --------------------------------- |
| 1    | torch.nn                                             | 2    | torch.nn.Module                   |
| 3    | torch.nn.ModuleList                                  | 4    | torch.nn.Sequential               |
| 5    | torch.nn.utils                                       | 6    | torch.nn.utils.clip_grad_value_   |
| 7    | torch.nn.Parameter                                   | 8    | torch.nn.DataParallel             |
| 9    | torch.nn.functional                                  | 10   | torch.nn.BatchNorm1d              |
| 11   | torch.nn.BatchNorm2d                                 | 12   | torch.nn.BatchNorm3d              |
| 13   | torch.nn.Conv1d                                      | 14   | torch.nn.Conv2d                   |
| 15   | torch.nn.Conv3d                                      | 16   | torch.nn.ConvTranspose2d          |
| 17   | torch.nn.Dropout                                     | 18   | torch.nn.Embedding                |
| 19   | torch.nn.InstanceNorm2d                              | 20   | torch.nn.LeakyReLU                |
| 21   | torch.nn.Linear                                      | 22   | torch.nn.MaxPool1d                |
| 23   | torch.nn.MaxPool2d                                   | 24   | torch.nn.MaxPool3d                |
| 25   | torch.nn.ReLU                                        | 26   | torch.nn.Sigmoid                  |
| 27   | torch.nn.Softmax                                     | 28   | torch.nn.Tanh                     |
| 29   | torch.nn.Upsample                                    | 30   | torch.nn.CrossEntropyLoss         |
| 31   | torch.nn.BCEWithLogitsLoss                           | 32   | torch.nn.BCELoss                  |
| 33   | torch.nn.functional.avg_pool1d                       | 34   | torch.nn.functional.avg_pool2d    |
| 35   | torch.nn.functional.avg_pool3d                       | 36   | torch.nn.functional.dropout       |
| 37   | torch.nn.functional.log_softmax                      | 38   | torch.nn.functional.pad           |
| 39   | torch.sigmoid                                        | 40   | torch.nn.functional.sigmoid       |
| 41   | torch.nn.functional.softmax                          | 42   | torch.nn.init.xavier_uniform_     |
| 43   | torch.nn.functional.binary_cross_entropy_with_logits | 44   | torch.nn.functional.cross_entropy |
| 45   | torch.nn.functional.dropout                          | 46   | torch.nn.functional.relu          |
| 47   | torch.nn.functional.smooth_l1_loss                   | 48     |    torch.nn.AdaptiveAvgPool1d      |
| 49   | torch.nn.AdaptiveAvgPool2d                | 50     |    torch.nn.AdaptiveAvgPool3d      |
| 51   | torch.nn.AvgPool1d                   | 52     |    torch.nn.AvgPool2d      |
| 53   | torch.nn.AvgPool3d                   | 54     |    torch.nn.ConstantPad2d      |
| 55   | torch.nn.Dropout2d                   | 56     |    torch.nn.GELU      |
| 57   | torch.nn.GroupNorm                  | 58     |    torch.nn.Identity      |
| 59   | torch.nn.LayerNorm                | 60     |    torch.nn.MaxUnpool2d      |
| 61   | torch.nn.ReflectionPad2d                   | 62     |    torch.nn.ReplicationPad2d      |
| 63   | torch.nn.PReLU                   | 64     |    torch.nn.SyncBatchNorm      |
| 65   | torch.nn.ZeroPad2d                   | 66     |    torch.nn.KLDivLoss      |
| 67   | torch.nn.L1Loss                  | 68     |    paddle.nn.functional.interpolate     |
| 69   | torch.nn.functional.mse_loss              |   70  |    torch.nn.init.constant_    |
| 71   | torch.nn.init.normal_                   | 72     |    torch.nn.init.ones_     |
| 73   | torch.nn.init.zeros_                  | 74     |    torch.nn.init.orthogonal_     |



## Utils类API

| 序号 | API                            | 序号 | API                         |
| ---- | ------------------------------ | ---- | --------------------------- |
| 1    | torch.utils.data               | 2    | torch.utils.data.DataLoader |
| 3    | torch.utils.data.random_split  | 4    | torch.utils.data.Dataset    |
| 5    | torch.utils.data.ConcatDataset | 6     | torch.utils.data.distributed      |
| 7    | torch.utils.data.distributed.DistributedSampler  | 8    | torch.utils.model_zoo |
| 9    | torch.utils.model_zoo.load_url  | 10    | torch.multiprocessing   |
| 11    | torch.multiprocessing.spawn              | 12    | torch.distributed |
| 13    | torch.distributed.init_process_group  | 14    |  |

## Autograd类API

| 序号 | API                     | 序号 | API                 |
| ---- | ----------------------- | ---- | ------------------- |
| 1    | torch.autograd.Variable | 2    | torch.autograd.grad |

## 基础操作API

| 序号 | API                     | 序号 | API                     |
| ---- | ----------------------- | ---- | ----------------------- |
| 1    | torch                   | 2    | torch.Tensor            |
| 3    | torch.FloatTensor       | 4    | torch.load              |
| 5    | torch.save              | 6    | torch.device            |
| 7    | torch.cat               | 8    | torch.cuda.is_available |
| 9    | torch.no_grad           | 10   | torch.from_numpy        |
| 11   | torch.cuda.device_count | 12   | torch.manual_seed       |
| 13   | torch.unsqueeze         | 14   | torch.squeeze           |
| 15   | torch.sum               | 16   | torch.mean              |
| 17   | torch.full              | 18   | torch.full_like         |
| 19   | torch.ones              | 20   | torch.ones_like         |
| 21   | torch.zeros             | 22   | torch.zeros_like        |
| 23   | torch.sqrt              | 24   | torch.arange            |
| 25   | torch.matmul            | 26   | torch.set_grad_enabled  |
| 27   | torch.tensor            | 28   | torch.clamp             |
| 29   | torch.exp               | 30   | torch.max               |
| 31   | torch.min               | 32   | torch.argmax            |
| 33   | torch.argmin            | 34   | torch.stack             |
| 35   | torch.log               | 36   | torch.randperm          |
| 37   | torch.rand              | 38   | torch.abs               |
| 39   | torch.bitwise_or        | 40   | torch.bitwise_xor       |
| 41   | torch.bitwise_and       | 42   | torch.bitwise_not       |
| 43   | torch.randn             | 44   | torch.add |
| 45   | torch.mul          | 46   | torch.linspace  |
| 47   | torch.einsum|     |        |

## Torchvision API

| 序号 | API                               | 序号 | API                                         |
| ---- | --------------------------------- | ---- | ------------------------------------------- |
| 1    | torchvision.transforms            | 2    | torchvision.transforms.Compose              |
| 3    | torchvision.transforms.ToPILImage | 4    | torchvision.transforms.Resize               |
| 5    | torchvision.transforms.ToTensor   | 6    | torchvision.transforms.RandomHorizontalFlip |
| 7    | torchvision.transforms.CenterCrop | 8    | torchvision.transforms.Normalize            |
| 9    | torchvision.utils.save_image      | 10   | torchvision.datasets.ImageFolder            |
| 11    | torchvision.transforms.RandomResizedCrop   | 12    | torchvision.transforms.Lambda              |
| 13    | torchvision.utils | 14    | torchvision.utils.save_image               |
| 15    | torchvision.datasets   | 16    | torchvision.datasets.ImageFolder |
| 17    | torchvision.models | 18    | torchvision.models.vgg_pth_urls           |
| 19    | torchvision.models.vgg11      | 20   | torchvision.models.vgg13            |
| 21    | torchvision.models.vgg16   | 22    | torchvision.models.vgg19             |
| 23    | torchvision.models.vgg11_bn | 24    | torchvision.models.vgg13_bn        |
| 25    | torchvision.models.vgg16_bn   | 26    | torchvision.models.vgg19_bn |
| 27    | torchvision.models.resnet34 | 28    | torchvision.models.resnet50       |
| 29    | torchvision.models.resnet101     | 30   | torchvision.models.resnet152            |
| 31    | torchvision.models.resnext50_32x4d   | 32    | torchvision.models.resnext101_32x8d       |
| 33    | torchvision.models.wide_resnet50_2 | 34    | torchvision.models.wide_resnet101_2      |


***持续更新...***