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 可以是标量或向量。向量输入必须全部具有相同的大小。标量输入被广播以匹配向量输入的大小。