#NEWLINE# @ application #NEWLINE# def generate( self , source_sentence , representation , **kwargs ) : #NEWLINE# #INDENT# return self . sequence_generator . generate ( n_steps = ( 2 * source_sentence . shape [ 1 ] ) , batch_size = source_sentence . shape [ 0 ] , attended = representation , attended_mask = tensor . ones ( source_sentence . shape ) . T , ** kwargs )