DataSet API Tensorflow를 사용하면 "None"차원으로 인해 오류가 발생합니다.

Question

Datetet API를 사용하여 최신 Tensorflow official models release에있는 resnet을 공급하려고합니다. 다음

기본 코드는 위로 최종 층은 조밀 한 층이다 resent_v2 정의를 참조한다

ValueError: The last dimension of the inputs to Dense should be defined. Found None .

컴파일시

with tf.Session() as sess: 
    print("initialized") 

    features_placeholder = tf.placeholder(prepared_x.dtype, prepared_x.shape) 
    labels_placeholder = tf.placeholder(dtype=tf.float32, shape=prepared_t.shape) 

    dataset = tf.contrib.data.Dataset.from_tensor_slices((features_placeholder, labels_placeholder)) 
    dataset = dataset.shuffle(buffer_size=10000) 
    dataset = dataset.batch(batch_size) 
    dataset = dataset.repeat(num_epoch) 

    iterator = dataset.make_initializable_iterator() 

    (next_x_test, next_t_test) = iterator.get_next() 
    next_x_test = tf.to_float(next_x_test, name='ToFloat') 


    sess.run(iterator.initializer, feed_dict={features_placeholder: prepared_x, 
               labels_placeholder: prepared_t}) 


    print(next_x_test) 
    print(next_t_test) 

    model = resnet_v2(resnet_size=50, num_classes=num_bins) 

    output = model(next_x_test,is_training=True) 

이 마지막 라인에서 에러가 발생.

내 기능 텐서의 모양을 어떻게 주장 할 수 있습니까?

Answer 1

tensor.set_shape을 사용하면 정의되지 않은 경우 텐서의 모양을 설정할 수 있습니다.