Присоединение различных 1D-тензоров к 3-му измерению 3D-тензоров

Question

def to_rnn_embed_tensor(tensor: Tensor, cond: list, letter_count=LETTERS_COUNT, cond_to_tensor_func = embed_line_to_tensor) -> list:
    input_sz, batch_sz = tensor.shape
    ret = torch.zeros(input_sz, batch_sz, letter_count)
    for i in range(input_sz):
        for j in range(batch_sz):
            cond_j = cond[j]
            cond_tensor = cond_to_tensor_func(cond_j)
            ret[i, j, int(tensor[i, j])] = 1
    return ret

Итак, у меня есть список строк, которые преобразуются в тензоры 1x7, которые я пытаюсь добавить в третье измерение. Поэтому ret должен быть input_sz x batch_sz x letter_count + 7. Проблема в том, что тензор, добавляемый к третьему измерению в точке j, должен соответствовать j-му индексу в списке cond.

По существу cond_tensor должен добавляться к ret [i, j] для всех j