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, то выходом будет плотный булев тензор [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. |
© 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