假设我有一个向量v和一个矩阵M - 它们不是输入,而是由其他操作产生的中间张量。我想进行逐列操作 - 即为M的每一列应用函数f。即
操作(v,M)= [f(v,M [:,i])],i = 1 .. N。
我所做的是以下(伪代码):
for i range(K.int_shape(M))[1]:
column = Lambda(lambda x:x[:,i])(M)
... <a code performing f(v, column)> ...
我发现,在运行期间(即model.fit())列在for循环中是相同的,而不是检索每列。我怀疑Lambda(lambda x:x [:,i])不是在Keras中检索列的正确方法。
在Keras的矩阵中进行逐列操作的正确方法是什么?