Python tf.ragged.range用法及代碼示例

返回包含指定數字序列的RaggedTensor。

用法

tf.ragged.range(
    starts, limits=None, deltas=1, dtype=None, name=None,
    row_splits_dtype=tf.dtypes.int64
)

返回的 RaggedTensor 的每一行都包含一個序列：

ragged.range(starts, limits, deltas)[i] ==
    tf.range(starts[i], limits[i], deltas[i])

如果 start[i] < limits[i] and deltas[i] > 0 ，那麽 output[i] 將是一個空列表。同樣，如果 start[i] > limits[i] and deltas[i] < 0 ，那麽 output[i] 將是一個空列表。此行為與 Python range 函數一致，但與 tf.range 操作不同，後者在這些情況下返回錯誤。

例子：

tf.ragged.range([3, 5, 2]).to_list()
[[0, 1, 2], [0, 1, 2, 3, 4], [0, 1]]
tf.ragged.range([0, 5, 8], [3, 3, 12]).to_list()
[[0, 1, 2], [], [8, 9, 10, 11]]
tf.ragged.range([0, 5, 8], [3, 3, 12], 2).to_list()
[[0, 2], [], [8, 10]]

輸入張量 starts , limits 和 deltas 可以是標量或向量。向量輸入必須全部具有相同的大小。標量輸入被廣播以匹配向量輸入的大小。