利用pytorch进行简单的卷积操作
input – input tensor of shape (/text{minibatch} , /text{in/_channels} , iH , iW)(minibatch,in_channels,iH,iW) weight – filters of shape (/text{out/_channels} , /frac{/text{in/_channels}}{/text{groups}} , kH , kW)(out_channels,groupsin_channels,kH,kW) 进行简单的卷积操作的使用:stride为1和2的时候他的输出是不一样的。 import torch
from torch import nn
import torch.nn.functional as F
input = torch.tensor([[1,2,0,3,1],
[0,1,2,3,1],
[1,2,1,0,0],
共有 0 条评论