Question

我有一个自定义的Keras图层，我使用Theano作为后端，我想进行以下操作：

假设我们有一个形状（N，）的张量。我想将K个第一个值设置为固定值x（3或其他......）。我怎么做？我假设我必须使用argsort，但我不知道如何使用Theano实现它。

例如，在简单的FF层中，如何将张量a的前N个值设置为值x？

def call(self, x, mask=None):
    a = K.dot(x, self.W)

    if self.bias:
        a += self.b

    return a

例如使用numpy我可以将a的10个第一个值设置为1，如下所示：

a[a.argsort() <= 10] = 1

我想使用Keras后端函数或Theano函数来做同样的事情。

Answer 1

对于a[a.argsort() <= 10] = 1，等效代码为：

import numpy as np
from keras import backend as K

a = np.asarray([[3,4,2,44,22,4,5,6,77,86,3,2,3,23,44,21],
                [3,4,22,44,2,4,54,6,77,8,3,2,36,23,4,2]], dtype=np.float)
a_t = K.variable(a)

a[a.argsort() <= 10] = 1
print a

arg_sort = K.T.argsort(a_t)
_cond = K.lesser_equal(arg_sort,10)
a_new = K.switch(_cond, 1, a_t)
print K.eval(a_new)

在一个声明中它将是：

a_new = K.switch(K.lesser_equal(K.T.argsort(a_t),10), 1, a_t)
print K.eval(a_new)

我不确定这是否正是您所需要的。

Theano / Keras：将Tensor的K-first值设置为一个值

1 个答案: