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. |
© 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/versions/r2.9/api_docs/python/tf/sparse/to_indicator