0
点赞
收藏
分享

微信扫一扫

Torch SpatialZeroPadding 实例

木樨点点 2022-08-01 阅读 139


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


举报

相关推荐

0 条评论