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
и False в остальных позициях 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, то выходным значением будет плотный булевый тензор со значением False во всех позициях, кроме
(0, 0, 0), (0, 1, 10), (1, 0, 103), (1, 1, 149), (1, 1, 150), (1, 2, 121).
Обратите внимание, что повторы в входном SparseTensor разрешены. Этот оператор полезен для преобразования SparseTensor в плотные форматы для совместимости с операторами, ожидающими плотные тензоры.
Входной SparseTensor должен быть в порядке строк.
| Аргументы | |
|---|---|
sp_input | 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/r1.15/api_docs/python/tf/sparse/to_indicator