#NEWLINE# @ application #NEWLINE# def initial_states( self , batch_size , * args , **kwargs ) : #NEWLINE# #INDENT# attended = kwargs [ 'attended' ] #NEWLINE# initial_state = self . initial_transformer . apply ( attended [ 0 , : , ( - self . attended_dim ) : ] ) #NEWLINE# return initial_state