torch向量/矩阵维度填充

达达

一、针对二维矩阵

所需库函数

import numpy as np

a = np.array([4,3,2,1,2,5,6,7])

矩阵维度

填充pad，在后面填充2个0

d = np.pad(a,(0,2),'constant')

填充后矩阵维度

二、针对三维矩阵

所需库函数

import torch

初始化矩阵

a = torch.ones(8,30,30)

原始矩阵

填充，在第二维和第三位分别填充2列0

b = F.pad(a,[0,2,0,2])

矩阵填充后

三、针对矩阵复制（横向复制）

生成（2,5）维度的矩阵，目的是复制成（2,5,5）的矩阵，其一维的每个维度复制5遍。

adj_score = torch.rand(2,5)

矩阵维度初始化

维度复制

adj_score_xlen_max =adj_score.detach().unsqueeze(dim=1).repeat(1,5,1)

image.png

四、针对矩阵复制（纵向复制）

初始化（2,5）的二维矩阵，将其复制成（2,5,5）的三维矩阵，其第三维的维度转成横向，复制5遍；

adj_score = torch.rand(2,5)
print(adj_score)
print(adj_score.shape)

初始化矩阵

adj_score = adj_score.unsqueeze(dim=2)
adj_score.shape
adj_score_xlen_max =adj_score.detach().repeat(1,1,5)
adj_score_xlen_max.shape
adj_score_xlen_max

结果