torch常用操作—— torch.nn.functional.pad
torch.nn.functional.pad作用:填充tensor目前为止,只支持2d和3d填充用法: torch.nn.functional.pad(input, pad, mode='constant', value=0)[source]input需要填充的tensorpad决定填充的方向以下面的例子观察可知:pad为tuple (pad_l, pad_r, pad_t, ...
原创
2020-01-04 13:51:15 ·
1816 阅读 ·
1 评论