从正常张量制作参差不齐的张量

Question

如果我有这样的张量

[[[1, 2], [2, 6], [7, 8]],
 [[7, 9], [1, 6], [3, 2]],
 [[8, 3], [3, 6], [3, 9]]]

和另一个张量，该张量指定必须从第一行开始的行的长度，例如像这样的[2, 3, 1]我该如何将第一个法线张量转换为这个参差不齐的张量：

[[[1, 2], [2, 6]],
 [[7, 9], [1, 6], [3, 2]],
 [[8, 3]]]

这能以不停止梯度反向传播的方式完成吗？