【Keras】K.switch()用法

本文主要是介绍【Keras】K.switch()用法，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

一、switch

keras.backend.switch(condition, then_expression, else_expression)

根据一个标量值在两个操作之间切换。

请注意，then_expression 和 else_expression 都应该是相同尺寸的符号张量。

参数

condition: 张量 (int 或 bool)。
then_expression: 张量或返回张量的可调用函数。
else_expression: 张量或返回张量的可调用函数。

返回

选择的张量。

switch接口，顾名思义，就是一个if/else条件判断语句。不过要求输入和输出都必须是张量。

 

二、 example

    import keras.backend as Kpred = K.placeholder([], dtype='bool')count = K.variable(0)x = K.switch(condition=pred,then_expression=lambda: K.update_add(count, 1),else_expression=lambda: K.update_add(count, 0))f = K.function(inputs=[pred], outputs=[x])print(f([True]))print(f([False]))# print [0.0][1.0]

 

这篇关于【Keras】K.switch()用法的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！

---