tf.sparse.to_indicator
| Просмотреть исходный код на GitHub |
Преобразует SparseTensor идентификаторов в плотный булевый индикаторный тензор.
tf.sparse.to_indicator(
sp_input, vocab_size, name=None
)
Последнее измерение sp_input.indices отбрасывается и заменяется значениями sp_input. Если sp_input.dense_shape = [D0, D1, ..., Dn, K], то output.shape = [D0, D1, ..., Dn, vocab_size], где
output[d_0, d_1, ..., d_n, sp_input[d_0, d_1, ..., d_n, k]] = True
и Ложь в остальных местах output.
Например, если sp_input.dense_shape = [2, 3, 4] с ненулевыми значениями:
[0, 0, 0]: 0 [0, 1, 0]: 10 [1, 0, 3]: 103 [1, 1, 1]: 150 [1, 1, 2]: 149 [1, 1, 3]: 150 [1, 2, 1]: 121
и vocab_size = 200, то выходной будет плотный булевый тензор с Ложью повсюду, кроме позиций
(0, 0, 0), (0, 1, 10), (1, 0, 103), (1, 1, 149), (1, 1, 150), (1, 2, 121).
Обратите внимание, что повторения разрешены в входном SparseTensor. Этот оператор полезен для преобразования SparseTensor в плотные форматы для совместимости с операторами, которые ожидают плотные тензоры.
Входной SparseTensor должен быть в строчно-главном порядке.
| Аргументы | |
|---|---|
sp_input | A SparseTensor с свойством values типа int32 или int64. |
vocab_size | Скалярный тензор int64 (или целое число Python), содержащий новый размер последнего измерения, all(0 <= sp_input.values < vocab_size). |
name | Префикс имени для возвращаемых тензоров (необязательно) |
| Возвращает | |
|---|---|
| Плотный булевый индикаторный тензор, представляющий индексы со значениями. |
| Возбуждает | |
|---|---|
TypeError | Если sp_input не является SparseTensor. |
© 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.4/api_docs/python/tf/sparse/to_indicator