@register_specialize
@register_canonicalize
@register_useless
@gof.local_optimizer([T.Join])
def local_join_1(node):
 if (not isinstance(node.op, T.Join)):
  return
 tensors = node.inputs[1:]
 if (len(tensors) == 1):
  return [tensors[0]]

@ register_specialize @ register_canonicalize @ register_useless @ gof . local_optimizer ( [ T . Join ] ) def local_join_1 ( node ) : if ( not isinstance ( node . op , T . Join ) ) : return tensors = node . inputs [ 1 : ] if ( len ( tensors ) == 1 ) : return [ tensors [ 0 ] ]
join => x remove join() when only one element is joined .