Факел: как вставить тензор в другой тензор в определенном индексе ⇐ Python

[Anonymous]

Я имею мягкий тензор x с формой (b, t1, c) и мягким тензором y с формой (b, t2, c) , я также знаю длину образца l для x . Я хочу вставить образцы x в y при определенном индексе i и накладке в конце.

Код: Выделить всё

inputs = []
for i in range(X.shape[0]):
input = torch.cat([Y[i][0:5],
X[i][:L[i]],
Y[i][5:],
torch.zeros(max(L) - L[i], Y.shape[2])],
dim=0)
inputs.append(input)
outputs = torch.stack(inputs, dim=0)

Я хочу знать, как сделать это в тензорной версии вместо цикла, которая слишком медленная для обучения.

Подробнее здесь: https://stackoverflow.com/questions/796 ... tain-index