code: def reshape(attrs, inputs, proto_obj):
      
      if len(inputs) == 1:
          return 'reshape', attrs, inputs[0]
      reshape_shape = list(proto_obj._params[inputs[1].name].asnumpy())
      reshape_shape = [int(i) for i in reshape_shape]
      new_attrs = {'shape': reshape_shape}
      return 'reshape', new_attrs, inputs[:1]
docs: Reshape the given array by the shape attribute.