Условное присвоение tf.variable в Tensorflow 2

Для numpy у нас есть

threshold = 3
a = np.array([1,2,3,4,5,6])
a[a>=3] = 199

# a is [1, 2, 199, 199, 199, 199]

Как написать аналогичный код в tensorflow 2

b = tf.Variable(a)

Спасибо.

python python-3.x conditional-statements tensorflow2.0

25.12.2020 17:05

Почему в Python есть оператор "pass"?

Оператор pass в Python - это простая концепция, которую могут быстро освоить даже новички без опыта программирования.

Некоторые методы, о которых вы не знали, что они существуют в Python

Python - самый известный и самый простой в изучении язык в наши дни. Имея широкий спектр применения в области машинного обучения, Data Science,...

Основы Python Часть I

Вы когда-нибудь задумывались, почему в программах на Python вы видите приведенный ниже код?

LeetCode - 1579. Удаление максимального числа ребер для сохранения полной проходимости графа

Алиса и Боб имеют неориентированный граф из n узлов и трех типов ребер:

Оптимизация кода с помощью тернарного оператора Python

И последнее, что мы хотели бы показать вам, прежде чем двигаться дальше, это

Советы по эффективной веб-разработке с помощью Python

Как веб-разработчик, Python может стать мощным инструментом для создания эффективных и масштабируемых веб-приложений.

361

Перейти к ответу Данный вопрос помечен как решенный

Ответы 1

Ответ принят как подходящий

Конечно, вы можете использовать tf.where для условной установки значений:

b = tf.Variable(a)
tf.where(b >= 3, 199, b)
# <tf.Tensor: shape=(6,), dtype=int64, numpy=array([  1,   2, 199, 199, 199, 199])>

Привет @ cs95, спасибо за ответ. Будет ли tf.where(b>=3, 199, b) подключаться к вычислительному графу и влиять ли на градиент b при обратном распространении? Спасибо,

— 25.12.2020 17:35

@jason Не могу сказать точно, извините. Возможно, вы могли бы открыть дополнительный вопрос.

— 25.12.2020 17:37

Конечно, проблема сейчас в этом посте stackoverflow.com/questions/65449945/…

— 25.12.2020 17:45

@jason Удачи с вопросом, я посмотрю, смогу ли я что-нибудь покопать.

— 25.12.2020 17:46

25.12.2020 17:22