【日常笔记】pytorch深度学习常用函数

  1. torch.nn
    在实际使用中,最常见的做法是继承nn.Module,撰写自己的网络层。nn.Module
  2. torch.nn.functional.pad()
    padding操作
    padding操作是给图像外围加像素点。
    具体:pytorch 中pad函数toch.nn.functional.pad()的使用
  3. detach()
    将variable参数从网络中隔离开,不参与参数更新
  4. permute(dims)
    将tensor的维度换位,假设img的维度为(28,28,3), img.permute(2,0,1)得到一个(3,28,28)的tensor.
  5. torch.nn.Conv2d(in_channels, out_channels, kernel_size, stride=1, padding=0, dilation=1, groups=1, bias=True)
    卷积。

你可能感兴趣的:(深度学习)