77d86929ba8c3fd29d70742764298e434b297298,pymanopt/autodiff/backends/_tensorflow.py,_TensorFlowBackend,compile_function,#_TensorFlowBackend#Any#Any#,48

Before Change

    @Backend._assert_backend_available
    def compile_function(self, objective, argument):
        if not isinstance(argument, list):
            def func(x):
                feed_dict = {argument: x}
                return self._session.run(objective, feed_dict)
        else:
            def func(x):
                feed_dict = {i: d for i, d in zip(argument, x)}
                return self._session.run(objective, feed_dict)

        return func

    @Backend._assert_backend_available
    def compute_gradient(self, objective, argument):

After Change

    @Backend._assert_backend_available
    def compile_function(self, function, arguments):
        flattened_arguments = flatten_arguments(arguments)
        if len(flattened_arguments) == 1:
            def unary_function(x):
                (argument,) = flattened_arguments
                feed_dict = {argument: x}
                return self._session.run(function, feed_dict)
            return unary_function

        def nary_function(xs):
            flattened_input = flatten_arguments(xs)
            feed_dict = {
                argument: x

Instances