我正在编码一个名为A的keras层,我需要两个张量a,b。我应该如何固定输入[a,b]以获得正确的回报(A的回报也是拖车)?
def get_model(latent_dim):
# Input variables
u = Input(shape=(k,), dtype='float32', name = 'u]')
i = Input(shape=(k,), dtype='float32', name = 'i]')
LA=A(latent_dim)
list = A([u ,i])
u_return = list_co[-2]
i_return = list_co[-1]
编码环境:简单来说,我将A设置为流:
A类(潜影): def调用(输入): m =输入[0] n =输入[1] 返回[m,n]
但还是不对
文件_add_inbound_node中的行“ C:\ ProgramData \ Anaconda3 \ envs \ network \ lib \ site-packages \ keras \ engine \ topology.py”,第703行 output_tensors [i] ._ keras_shape = output_shapes [i] IndexError:列表索引超出范围