如何在张量流中使用if语句?

时间:2018-05-16 18:25:56

标签: python tensorflow

我认为我必须以不同的方式提出我的问题。我想这样做完全如下: 我在keras层中使用了relu(x)函数,现在我想改变它并创建我自己的激活函数,如下所示: 1.将数组创建为一个查找表,其值在[-5,5]和's'之间(s = 0.1),所以我的查找表应该包含102个数字,例如。 [-5,-4.9 ... + 4.9,5] 2.在上面查找表的形状中创建Y数组.Y的初始值为零,{Yi}是训练中可学习的参数 3 =计算f(x)= 1 / s(yi(xi + 1-x)+ yi + 1(x-xi)),如果xi≤x≤xi+ 1  对于xi和xi + 1之间的任何输入值,输出为  从yi和yi + 1线性插值,xi和xi + 1应该在查找表中找到,相应的yi和yi + 1在Y中找到 4 = RELU(F(X)) 问题是如何在keras中编写这个激活函数?

1 个答案:

答案 0 :(得分:0)

如果这是numpy,只需使用逻辑索引。

g = LT[LT > number]
l = LT[LT <= number]