Python tf.raw_ops.RaggedRange用法及代碼示例

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

用法

tf.raw_ops.RaggedRange(
    starts, limits, deltas, Tsplits=tf.dtypes.int64, name=None
)

參數

starts 一個Tensor。必須是以下類型之一：bfloat16 , float32 , float64 , int32 , int64。每個範圍的開始。
limits 一個Tensor。必須與 starts 具有相同的類型。每個範圍的限製。
deltas 一個Tensor。必須與 starts 具有相同的類型。每個範圍的增量。
Tsplits 一個可選的 tf.DType 來自：tf.int32, tf.int64。默認為 tf.int64 。
name 操作的名稱(可選)。

Tensor 對象的元組(rt_nested_splits、rt_dense_values)。
rt_nested_splits Tensor 類型為 Tsplits 。
rt_dense_values 一個Tensor。具有與 starts 相同的類型。

返回由 rt_dense_values 和 rt_nested_splits 組成的 RaggedTensor result ，例如 result[i] = range(starts[i], limits[i], deltas[i]) 。

(rt_nested_splits, rt_dense_values) = ragged_range(
      starts=[2, 5, 8], limits=[3, 5, 12], deltas=1)
result = tf.ragged.from_row_splits(rt_dense_values, rt_nested_splits)
print(result)
<tf.RaggedTensor [[2], [], [8, 9, 10, 11]] >

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

相關用法

注：本文由純淨天空篩選整理自tensorflow.org大神的英文原創作品 tf.raw_ops.RaggedRange。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。