parameters: (a, e, s0)
return_statement: return out
docstring: Build 3D model
docstring_summary: Build 3D model
docstring_tokens: Build 3D model
function: def build_model(a, e, s0):
          """Build 3D model"""
          assert (s0.shape[1] == 3)
          assert (e.shape[2] == 3)
          assert (a.shape[1] == e.shape[1])
          out = np.einsum('...i,...ijk', a, e)
          out += s0
          return out
function_tokens: def build_model ( a , e , s0 ) : assert ( s0 . shape [ 1 ] == 3 ) assert ( e . shape [ 2 ] == 3 ) assert ( a . shape [ 1 ] == e . shape [ 1 ] ) out = np . einsum ( '...i,...ijk' , a , e ) out += s0 return out