Skip to content

Is __call__ in ntm_cell correct? #21

@lsy643

Description

@lsy643

Hi
I am some kind of confused about the implementation of ntm_cell, especially the call() part.
According to the code, it seems that the new_output has nothing to do with the memory part.
Should it be more plausible if the new_output adapts itself with memory part?
`
output_list, hidden_list = self.build_controller(input_, read_list_prev,
output_list_prev,
hidden_list_prev)

    # last output layer from LSTM controller
    last_output = output_list[-1]

    # build a memory
    M, read_w_list, write_w_list, read_list = self.build_memory(M_prev,
                                                                read_w_list_prev,
                                                                write_w_list_prev,
                                                                last_output)

    # get a new output
    new_output, new_output_logit = self.new_output(last_output)

`

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions