Dimension must equal, MatMul with input shapes: [?,53],[150,200]

发布于 2022-09-12 01:28:12 字数 1636 浏览 22 评论 0

我想将两个不同维度的向量送进Bi-LSTM中,然后concat起来,但是报错。

with tf.variable_scope("bi-lstm"):
    cell_bw = LSTMCell(self.hidden_dim)
    cell_fw = LSTMCell(self.hidden_dim)

    (output_fw_seq_e, output_bw_seq_e), _ = tf.nn.bidirectional_dynamic_rnn(
        cell_fw=cell_fw,
        cell_bw=cell_bw,
        inputs=self.word_embeddings,
        sequence_length=self.sequence_lengths,
        dtype=tf.float32)
    output_e = tf.concat([output_fw_seq_e, output_bw_seq_e], axis=-1)

    (output_fw_seq_d, output_bw_seq_d), _ = tf.nn.bidirectional_dynamic_rnn(
        cell_fw=cell_fw,
        cell_bw=cell_bw,
        inputs=self.features_dim,
        sequence_length=self.sequence_lengths,
        dtype=tf.float32)
    output_d = tf.concat([output_fw_seq_d, output_bw_seq_d], axis=-1)

    output = tf.concat([output_e, output_d], axis=-1)
    output = tf.nn.dropout(output, self.dropout_pl)

self.hidden_dim: 50维
self.word_embeddings: 100维
self.features: 3维

求助!
补充Weight和bias代码如下:

with tf.variable_scope("proj"):
    W = tf.get_variable(name="W",
                        shape=[2 * self.hidden_dim, self.num_tags],
                        initializer=tf.contrib.layers.xavier_initializer(),
                        dtype=tf.float32)

    b = tf.get_variable(name="b",
                        shape=[self.num_tags],
                        initializer=tf.zeros_initializer(),
                        dtype=tf.float32)

    s = tf.shape(output)
    output = tf.reshape(output, [-1, 2 * self.hidden_dim])
    pred = tf.matmul(output, W) + b

如果你对这篇内容有疑问,欢迎到本站社区发帖提问 参与讨论,获取更多帮助,或者扫码二维码加入 Web 技术交流群。

扫码二维码加入Web技术交流群

发布评论

需要 登录 才能够评论, 你可以免费 注册 一个本站的账号。

评论(1

紫竹語嫣☆ 2022-09-19 01:28:12

[?,53][150,200]不能矩阵乘,第一个矩阵的第二维和第二个矩阵的第一维要相同。
问题应该是上一步的输出不对,或者模型不对。

试一下生成网络结构图,效果如下,http://ethereon.github.io/net...
20190323220419900.png

~没有更多了~
我们使用 Cookies 和其他技术来定制您的体验包括您的登录状态等。通过阅读我们的 隐私政策 了解更多相关信息。 单击 接受 或继续使用网站,即表示您同意使用 Cookies 和您的相关数据。
原文