LSTM 返回状态

def gru_model(units1=[9,196,196*2,196*4,196]):

    inputs=Input(shape=(units1[0],units1[1]))

    lstm=GRU(units=units1[2],return_sequences=True)(inputs)

    lstm=Dropout(rate=0.5)(lstm)

    lstm=GRU(units=units1[3],return_sequences=True)(lstm)

    lstm=Dropout(rate=0.5)(lstm)

    lstm=GRU(units=units1[4],return_sequences=True,return_state=True)(lstm)

#  return_sequences=true 返回整个序列 也就是输入n个时间步 设定uints=m 返回[none,n,m]

#  return_sequences=false  只返回最后一个状态  设定uints=m 返回[none,m]

#  return_state=True  返回最后一个状态 与 return_sequences=false 返回相同 [[],[]]

#  return_state=True  只返回序列 无最后状态



    outputs=lstm

你可能感兴趣的:(LSTM 返回状态)