pytorch对n2维度的tensorpad第一个维度到1024

可以使用torch.nn.functional.pad函数来实现对n*2维度的tensor进行pad操作。

import torch
import torch.nn.functional as F

# 假设输入的tensor是x，形状为(n, 2)
x = torch.randn(n, 2)

# pad的目标长度为1024
target_length = 1024

# 计算需要pad的长度
pad_length = target_length - x.size(0)

# 使用pad函数进行pad操作，对第一个维度进行pad，pad的长度为pad_length
padded_x = F.pad(x, (0, 0, 0, pad_length))

# 输出padded_x的形状
print(padded_x.size())  # 输出：torch.Size([1024, 2])

这样就可以将输入的tensor的第一个维度pad到目标长度1024。