CODEWITHSUNDEEP
pythontensorflowexception
Yunfeng Zhao
Yunfeng Zhao

Reputation: 73

Tensorflow UnimplementedError

I have implemented the following example in Tensorflow:

import tensorflow as tf
import numpy as np


def loss_function(values, a, b):
    N = values.shape[0]
    i = tf.constant(0)
    values_array = tf.get_variable(
        "values", values.shape, initializer=tf.constant_initializer(values))
    result = tf.constant(0, dtype=tf.float32)

    def body1(i):

        op2 = tf.assign(values_array[i, 0],
                        a + b)  # Here is where it should be updated. The value being assigned is actually calculated from variable a and b.

        with tf.control_dependencies([op2]):
            return tf.identity(i + 1)

    def condition1(i): return tf.less(i, N)
    i = tf.while_loop(condition1, body1, [i])

    with tf.control_dependencies([i]):
        op1 = tf.assign(values_array[0, 0],
                        999.0)  # Here is where it should be updated

        # The final cost is calculated based on the entire values_array
        with tf.control_dependencies([op1]):
            result = result + tf.reduce_mean(values_array)
            return tf.identity(result)


# The parameters we want to calculate in the end
a = tf.Variable(tf.random_uniform([1], 0, 700), name='a')
b = tf.Variable(tf.random_uniform([1], -700, 700), name='b')

values = np.ones([2, 4], dtype=np.float32)

# cost function
cost_function = loss_function(values, a, b)

# training algorithm
optimizer = tf.train.MomentumOptimizer(
    0.1, momentum=0.9).minimize(cost_function)

# initializing the variables
init = tf.global_variables_initializer()

# starting the session session
sess = tf.Session()
sess.run(init)

training_cost = sess.run([cost_function])
_ = sess.run([optimizer])

print tf.get_collection(
    tf.GraphKeys.GLOBAL_VARIABLES, scope="values")[0].eval(session=sess)

Overall, what I want is that the cost function calculates a temporary 2D array based on the input numpy 2D array and parameters a and b. Then, the final cost is calculated from the temporary 2D array. However, it throws a Exception has occurred: UnimplementedError UnimplementedError() exception.

Any help?

Thanks!

Upvotes: 1

Views: 1997

Answers (1)

Yunfeng Zhao
Yunfeng Zhao

Reputation: 73

I figured it out. It throws UnimplementedError (see above for traceback): sliced l-value shape [] does not match r-value shape [1]. Automatic broadcasting not yet implemented.. Therefore, change op2 = tf.assign(values_array[i, 0], a + b) to op2 = tf.assign(values_array[i, 0], (a + b)[0]) will solve this issue.

Upvotes: 4

Related Questions