簡體   English   中英

應用密集層后的尺寸誤差

[英]Dimensionality error after applying a dense layer

我在對最大池化卷積層輸出應用dropout之后,嘗試添加一個密集層。

我有以下用Python編寫的TensorFlow代碼。 過濾器數為128,len(filter_sizes)為3

pooled_outputs = []
    for i, filter_size in enumerate(filter_sizes):
        with tf.name_scope("conv-maxpool-%s" % filter_size):

            # Convolution Layer
            filter_shape = [filter_size, embedding_size, 1, num_filters]
            W = tf.Variable(tf.truncated_normal(filter_shape, stddev=0.1), name="W")
            b = tf.Variable(tf.constant(0.1, shape=[num_filters]), name="b")
            conv = tf.nn.conv2d(
                self.embedded_chars_expanded,
                W,
                strides=[1, 1, 1, 1],
                padding="VALID",
                name="conv")

            # Applying batch normalization
            # h = tf.contrib.layers.batch_norm(conv, center=True, scale=True, is_training=True)

            # Apply nonlinearity
            h1 = tf.nn.relu(tf.nn.bias_add(conv, b), name="relu")

            # Maxpooling over the outputs
            pooled = tf.nn.max_pool(
                h1,
                ksize=[1, sequence_length - filter_size + 1, 1, 1],
                strides=[1, 1, 1, 1],
                padding='VALID',
                name="pool")
            pooled_outputs.append(pooled)

    # Combine all the pooled features
    num_filters_total = num_filters * len(filter_sizes)
    self.h_pool = tf.concat(pooled_outputs, 3)
    self.h_pool_flat = tf.reshape(self.h_pool, [-1, num_filters_total])

    # Add dropout
    with tf.name_scope("dropout"):
        #self.h_drop = tf.nn.dropout(dense, self.dropout_keep_prob)
        self.h_drop = tf.nn.dropout(self.h_pool_flat, self.dropout_keep_prob)

        # Adding dense layer
    dense = tf.layers.dense(self.h_drop, units=num_classes, activation=tf.nn.relu)

密集層應用后面臨的問題。

以下是錯誤:

Dimensions must be equal, but are 11 and 384 for 'output/scores/MatMul' (op: 'MatMul') with input shapes: [?,11], [384,11]

有人可以幫我嗎?

誤差在於矩陣的索引。 我正在使用由tensorflow提供的xw_plus_b函數,並且將矩陣的維數用於錯誤的乘法運算。

暫無
暫無

聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM