卷积神经网络:最大池化层

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
今日推荐