PyTorch_Conv2d
2D卷积
文章图片
代码
import torch
import torch.nn as nninput = torch.randn(1, 3, 112, 112)
conv2d = nn.Conv2d(3, 64, (3, 3), 2, 1)
output = conv2d(input)
print(output.size())torch.Size([1, 64, 56, 56])
引用 <1>
torch=1.7.1+cu101
torchvision=0.8.2
torchaudio=0.7.2
<2>
【PyTorch_Conv2d】https://pytorch.org/docs/1.7....
推荐阅读
- pytorch|使用pytorch从头实现多层LSTM
- Pytorch|Pytorch AlexNet Fashion-MNIST
- pytorch|YOLOX 阅读笔记
- Pytorch学习|sklearn-SVM 模型保存、交叉验证与网格搜索
- 轻量模块注意力机制ECA-Net(注意力模块+一维卷积)
- Keras|将Pytorch模型迁移到android端(android studio)【未实现】
- Android|将Pytorch模型部署到Android端
- nvidia|nvidia jetson xavier nx安装pytorch
- python|PyTorch单机多卡分布式训练教程及代码示例
- 深度瞎搞|PyTorch单机多卡训练(DDP-DistributedDataParallel的使用)备忘记录