tf.keras.ops.select
Возвращает элементы из choicelist на основе условий в condlist.
tf.keras.ops.select(
condlist, choicelist, default=0
)
| Аргументы | |
|---|---|
condlist | Список булевых тензоров. Список условий, определяющих, из какого массива в choicelist берутся выходные элементы. Если выполняется несколько условий, используется первое из них, встречающееся в condlist. |
choicelist | Список тензоров. Список тензоров, из которых берутся выходные элементы. Длина этого списка должна совпадать с длиной condlist. |
defaults | Необязательное скалярное значение. Элемент, добавляемый в выходной результат, если все условия принимают значение False. |
| Возвращает | |
|---|---|
Тензор, в котором выходной элемент в позиции m является m-м элементом тензора в choicelist, где m-й элемент соответствующего тензора в condlist равен True. |
Пример:
from keras import ops x = ops.arange(6) condlist = [x<3, x>3] choicelist = [x, x**2] ops.select(condlist, choicelist, 42) # Returns: tensor([0, 1, 2, 42, 16, 25])
© 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/keras/ops/select
