Python tf.reverse_sequence用法及代碼示例

反轉可變長度切片。

用法

tf.reverse_sequence(
    input, seq_lengths, seq_axis=None, batch_axis=None, name=None
)

此操作首先沿維度 batch_axis 切片 input ，並且對於每個切片 i ，沿維度 seq_axis 反轉第一個 seq_lengths[i] 元素。

seq_lengths 的元素必須服從 seq_lengths[i] <= input.dims[seq_axis] ，並且 seq_lengths 必須是長度為 input.dims[batch_axis] 的向量。

沿維度 batch_axis 的輸出切片 i 然後由輸入切片 i 給出，沿維度 seq_axis 的第一個 seq_lengths[i] 切片反轉。

示例用法：

seq_lengths = [7, 2, 3, 5]
input = [[1, 2, 3, 4, 5, 0, 0, 0], [1, 2, 0, 0, 0, 0, 0, 0],
         [1, 2, 3, 4, 0, 0, 0, 0], [1, 2, 3, 4, 5, 6, 7, 8]]
output = tf.reverse_sequence(input, seq_lengths, seq_axis=1, batch_axis=0)
output
<tf.Tensor:shape=(4, 8), dtype=int32, numpy=
array([[0, 0, 5, 4, 3, 2, 1, 0],
       [2, 1, 0, 0, 0, 0, 0, 0],
       [3, 2, 1, 4, 0, 0, 0, 0],
       [5, 4, 3, 2, 1, 6, 7, 8]], dtype=int32)>