torch中卷积层和其它层

import torch
import torch.nn as nn
import torch.nn.functional as F

详情可以参考官方文档https://pytorch-cn.readthedocs.io/zh/latest/package_references/torch-nn/#torchnn