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