小编Rit*_*max的帖子

我正在寻找类似于python "numpy.where()"命令的Keras 命令.基本上,我的想法是从张量中提取指数.在python我可以做的只是f_j=(np.where(X==j))让我具体indices(f_j)的价值j.

例如:

X= [0 1 1 0 0 2 3 ]

f_j=(np.where(X==1))

f_j= [1 2]

是否有任何类似的功能可用于此目的？

我试着在张量内编写数组搜索.但是,当调用"if K.equal():"line as 时,我最终会出错

TypeError:不允许使用tf.Tensor作为Python bool.如果t不是None,则使用:而不是if t:来测试是否定义了张量,并使用TensorFlow操作(如tf.cond)来执行以张量值为条件的子图.

def loss(y_true, y_pred:

b=K.equal(y_true,0)

b=K.cast(b,dtype='float32')

for i in range(0,5):

if K.equal(b[i],1):

........

y_true = [0 1 1 0 0 2 3 ]

根据我的理解,Keras LSTM细胞类似于Tensorflow

BasicLSTMCell(RNNCell).但是,我正在寻找Tensorflow的实现

Keras的LSTMCell(RNNCell) (LSTMCell:该类使用可选的窥孔连接,可选的单元裁剪和可选的投影层).https://github.com/tensorflow/tensorflow/blob/master/tensorflow/python/ops/rnn_cell_impl.py https://research.google.com/pubs/archive/43905.pdf Hasim Sak,Andrew Senior和Francoise Beaufays."用于大规模声学建模的长期短期记忆递归神经网络架构." INTERSPEECH,2014年.

我想知道没有编写自定义图层,是否有任何方法可以在Keras中使用此Tensorflow LSTMCell或者它是否已经可用？