Torch SpatialZeroPadding 实例

来源:互联网 发布:windows启动过程 编辑:程序博客网 时间:2024/05/21 07:48
th> a = torch.ones(3,2,2)th> b = nn.SpatialZeroPadding(0, 0, 1)(a)th> b

(1,.,.) =
0 0
1 1
1 1

(2,.,.) =
0 0
1 1
1 1

(3,.,.) =
0 0
1 1
1 1

th> b = nn.SpatialZeroPadding(0, 0, 2)(a)   th> b

(1,.,.) =
0 0
0 0
1 1
1 1

(2,.,.) =
0 0
0 0
1 1
1 1

(3,.,.) =
0 0
0 0
1 1
1 1

th> b = nn.SpatialZeroPadding(0, 1, 0)(a)th> b

(1,.,.) =
1 1 0
1 1 0

(2,.,.) =
1 1 0
1 1 0

(3,.,.) =
1 1 0
1 1 0

th> b = nn.SpatialZeroPadding(1, 0, 0)(a)th> b

(1,.,.) =
0 1 1
0 1 1
0 0 0

(2,.,.) =
0 1 1
0 1 1
0 0 0

(3,.,.) =
0 1 1
0 1 1
0 0 0

原创粉丝点击