<s> from . import NeuralLayer <EOL> class Combine ( NeuralLayer ) : <EOL> """<STR_LIT>""" <EOL> def __init__ ( self , func , dim = <NUM_LIT:0> ) : <EOL> """<STR_LIT>""" <EOL> super ( Combine , self ) . __init__ ( "<STR_LIT>" ) <EOL> self . func = func <EOL> if dim > <NUM_LIT:0> : <EOL> self . output_dim = dim <EOL> def prepare ( self ) : <EOL> if self . output_dim == <NUM_LIT:0> : <EOL> self . output_dim = self . input_dim <EOL> def compute_tensor ( self , * tensors ) : <EOL> return self . func ( * tensors ) <EOL> def compute_test_tesnor ( self , * tensors ) : <EOL> return self . func ( * tensors ) </s>
