import torch
input=[1,2,3,4,
6,7,8,9,
1,2,3,4,
6,7,8,9,]
input=torch.Tensor(input).view(1,1,4,4) #(b*c*h*w)
maxpooling_layer=torch.nn.MaxPool2d(kernel_size=2,
stride=2,) #默认步幅为2
output=maxpooling_layer(input)
print(output)
卷积神经网络:最大池化层
猜你喜欢
转载自blog.csdn.net/qq_21686871/article/details/114295414
今日推荐
周排行