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, то выход будет плотным булевым тензором [2, 3, 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/r2.3/api_docs/python/tf/sparse/to_indicator