我有形狀為 (batch_size, 200, 256) 的 LSTM 層的輸出，其中 200 是標記序列的長度，256 是 LSTM 輸出維度。我還有另一個形狀為 (batch_size) 的張量，它是我想從批次中的每個樣本序列中切出的標記的索引列表。如果令牌索引不是 -1，我將切出一個令牌向量表示（長度 = 256）。如果令牌索引為 -1，我將給出零向量（長度 = 256）。預期的輸出結果具有形狀 (batch_size, 1, 256)。我該怎么做？謝謝這是我到目前為止嘗試過的bidir = concatenate([forward, backward]) # shape = (batch_size, 200, 256) dropout = Dropout(params['dropout_rate'])(bidir)def slice_by_tensor(x):    matrix_to_slice = x[0]    index_tensor = x[1]    out_tensor = tf.where(index_tensor == -1,                           tf.zeros(tf.shape(tf.gather(matrix_to_slice,                                                       index_tensor, axis=1))),                           tf.gather(matrix_to_slice, index_tensor, axis=1))    return out_tensorrepresentation_stack0 = Lambda(lambda x: slice_by_tensor(x))([dropout,stack_idx0]) # stack_idx0 shape is (batch_size) # I got output with shape (batch_size, batch_size, 256) with this code