tf.sparse.to_indicator
Преобразует 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. |
© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/sparse/to_indicator