torch.nn.functional.pad

作用

用来对一个tensor进行填充。最典型的就是图片了，原来是2*2的，现在想要变成3*3的，那么就需要填充，此时有很多选择，例如是在原来的右上进行填充还是左下？又或者是左上？等等。

这个函数就可以用来实现这些功能。

torch.nn.functional.pad(input, pad, mode='constant', value=0.0)

我们只关心第二个参数pad,其他不要管。pad是一个元组，格式如下：

(1,1)表示在最后一个维度，前面填充1个0，后面填充1个0
(1,0)表示在最后一个维度，前面填充1个0，后面不填充。
(1,1,1,1)表示最后两个维度都要填充，并且上下左右各填充1个0。
.....

下面，我们根据上面举的3个案例进行实战：

实战

import torch
import torch.nn.functional as tnf

x = torch.tensor([[1, 2],[ 3,4]])

print(x)
print(tnf.pad(x,(1,1)))#案例1
print(tnf.pad(x,(1,0)))#案例2
print(tnf.pad(x,(1,1,1,1)))#案例3

那开头的2*2怎么填充为3*3的图片呢？假设我们在右边和下边各填充1个0，那么就是这样的：

print(x)
print(tnf.pad(x,(0,1,0,1)))

torch.nn.functional.pad相关推荐

【pytorch】torch.nn.functional.pad的使用
torch.nn.functional.pad 是对Tensor做padding,输入的参数必须的torch的Tensor 一般地,习惯上会做如下声明 import torch.nn.function ...
pytorch笔记：torch.nn.functional.pad
1 torch.nn.functional.pad函数 torch.nn.functional.pad是pytorch内置的tensor扩充函数,便于对数据集图像或中间层特征进行维度扩充 torch. ...
torch.nn.functional.pad(input, pad, mode=‘constant‘, value=0)
torch.nn.functional.pad(input, pad, mode='constant', value=0) 填充Tensor. 填充大小: 填充input的某些维度的填充大小从最后一个 ...
mindspore.ops.Pad如何像torch.nn.functional.pad一样可以填充负维度
torch.nn.functional.pad[x,[-1,-1,-1,-1,]]可以填充负维度,而mindspore.ops.Pad 却不能填充负维度. 但是在官网中写的torch.nn.funct ...
Pytorch之torch.nn.functional.pad函数详解
torch.nn.functional.pad是PyTorch内置的矩阵填充函数 (1).torch.nn.functional.pad函数详细描述如下: torch.nn.functional.pa ...
torch.nn.functional
1. 非线性激活函数 torch.nn.functional.threshold(input, threshold, value, inplace=False) torch.nn.functional ...
python语言中ch用法_pytorch 中pad函数toch.nn.functional.pad()的用法
padding操作是给图像外围加像素点. 为了实际说明操作过程,这里我们使用一张实际的图片来做一下处理. 这张图片是大小是(256,256),使用pad来给它加上一个黑色的边框.具体代码如下: imp ...
pytorch 中pad函数toch.nn.functional.pad()的使用
padding操作是给图像外围加像素点. 为了实际说明操作过程,这里我们使用一张实际的图片来做一下处理. 这张图片是大小是(256,256),使用pad来给它加上一个黑色的边框.具体代码如下: imp ...
torch.nn.functional.cross_entropy.ignore_index
ignore_index表示计算交叉熵时,自动忽略的标签值,example: import torch import torch.nn.functional as F pred = [] pred.a ...

torch.nn.functional.pad

torch.nn.functional.pad相关推荐

最新文章

热门文章