Python Pytork将一系列操作更改为一系列状态

我的pytorch模型以张量[x1，x2，x3，…x10]的形式输出动作，表示改变当前环境的动作序列。我想把张量换成a 使用填充值初始化空状态的状态序列，我将表示f

所以，我的目标是转化

O_1 = [x1, x2, x3, x4, x5, x6, x7, x8, x9, x10] 

进入

我可以手工编写一个函数来实现这一点，但我想知道是否有一种更简单的pytorch-y方法（比如某些张量变换）来实现这一转换。此外，有没有一种方法可以将计算图从O_1保留到O_2

O_2 = [
[x1, f, f, f, f, f, f, f, f, f],
[x1, x2, f, f, f, f, f, f, f, f],
[x1, x2, x3, f, f, f, f, f, f, f],
[x1, x2, x3, x4, f, f, f, f, f, f],
[x1, x2, x3, x4, x5, f, f, f, f, f],
[x1, x2, x3, x4, x5, x6, f, f, f, f],
[x1, x2, x3, x4, x5, x6, x7, f, f, f],
[x1, x2, x3, x4, x5, x6, x7, x8, f, f],
[x1, x2, x3, x4, x5, x6, x7, x8, x9, f],
[x1, x2, x3, x4, x5, x6, x7, x8, x9, x10]
]