tf.ragged.range
| Просмотреть исходный код на GitHub |
Возвращает 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 могут быть скалярами или векторами. Векторные входы должны иметь одинаковый размер. Скалярные входы транслируются для соответствия размеру векторных входов.
| Аргументы | |
|---|---|
starts | Вектор или скаляр Tensor. Указывает первое значение для каждого диапазона, если limits не None; в противном случае указывает границы диапазона, а первые значения по умолчанию равны 0. |
limits | Вектор или скаляр Tensor. Указывает исключительные верхние пределы для каждого диапазона. |
deltas | Вектор или скаляр Tensor. Указывает шаг для каждого диапазона. По умолчанию 1. |
dtype | Тип элементов результирующего тензора. Если не указано, значение выбирается на основе других аргументов. |
name | Имя операции. |
row_splits_dtype | dtype для возвращаемого RaggedTensor тензора row_splits. Один из tf.int32 или tf.int64. |
| Возвращаемое значение | |
|---|---|
RaggedTensor типа dtype с ragged_rank=1. |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.3/api_docs/python/tf/ragged/range