torch.nn.ConstantPad1d 填充矩阵

函数功能：通过填充指定值来扩展tensor的最后一个维度，输入可以为任意维度的tensor。

x维度为1时

import torch
from torch import nn

conpad = nn.ConstantPad1d(2, 3.5)
x = torch.ones(2)
print("------------初始矩阵------------")
print(x)
x1 = conpad(x)
print("------------nn.ConstantPad1d(2, 3.5)------------")
print(x1.shape)
print(x1)
conpad2 = nn.ConstantPad1d((5, 1), 3.5)
x2 = conpad2(x)
print("------------nn.ConstantPad1d((5, 1), 3.5)------------")
print(x2.shape)
print(x2)

 x维度为3时

import torch
from torch import nn

conpad = nn.ConstantPad1d(2, 3.5)
x = torch.ones(1, 2, 3)
print("------------初始矩阵------------")
print(x)
x1 = conpad(x)
print("------------nn.ConstantPad1d(2, 3.5)------------")
print(x1.shape)
print(x1)
conpad2 = nn.ConstantPad1d((5, 1), 3.5)
x2 = conpad2(x)
print("------------nn.ConstantPad1d((5, 1), 3.5)------------")
print(x2.shape)
print(x2)