Torch SpatialAveragePooling 实例

来源:互联网 发布:万能数据恢复大师 编辑:程序博客网 时间:2024/04/30 07:11
th> a = torch.ones(4,3,2)th> b = nn.SpatialAveragePooling(2,2)(a)th> b

(1,.,.) =
1
1

(2,.,.) =
1
1

(3,.,.) =
1
1

(4,.,.) =
1
1

th> a = torch.ones(3,2,2)th> b = nn.SpatialAveragePooling(2,2)(a)th> b

(1,.,.) =
1

(2,.,.) =
1

(3,.,.) =
1