首页
学习
活动
专区
工具
TVP
发布
社区首页 >问答首页 >ValueError:形状必须是等级2,但对于输入形状:[6],[6],'MatMul‘(op:'MatMul')的等级是1

ValueError:形状必须是等级2,但对于输入形状:[6],[6],'MatMul‘(op:'MatMul')的等级是1
EN

Stack Overflow用户
提问于 2019-02-15 14:35:13
回答 3查看 952关注 0票数 1

错误:

ValueError:形状的等级必须为2,但对于输入形状为6、6的“”MatMul“”(op:“”MatMul“”),等级为1。“

代码语言:javascript
复制
import tensorflow as tf

with tf.device('/gpu:1'):
    a = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], name='a')
    b = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], name='b')
    c = tf.matmul(a, b)

sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
print(sess.run(c))

我不知道出了什么问题。非常感谢你的帮助。

EN

回答 3

Stack Overflow用户

发布于 2019-02-18 15:19:25

tf.matmul将矩阵与2维张量相乘。你正试图用matmul乘以两个向量,它们是一维的张量。

您预期的结果是[ 1. 4. 9. 16. 25. 36.],这是向量元素的元素乘法。要获得它,您必须使用tf.multiply操作。

代码语言:javascript
复制
import tensorflow as tf

a = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], name="a")
b = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], name="b")
c = tf.multiply(a, b)

sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
print(sess.run(c))
票数 1
EN

Stack Overflow用户

发布于 2019-02-18 15:28:29

否则,如果您想要进行矩阵乘法,而不是像其他答案中建议的那样按元素进行乘法,则需要将向量为2D到行向量与列向量的乘积:

代码语言:javascript
复制
import tensorflow as tf
a = tf.constant([[1.0], [2.0], [3.0], [4.0], [5.0], [6.0]], name="a") # Shape [6, 1]
b = tf.constant([[1.0, 2.0, 3.0, 4.0, 5.0, 6.0]], name="b") # Shape [1, 6]
c = tf.matmul(a, b)
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True)) print(sess.run(c))
票数 0
EN

Stack Overflow用户

发布于 2019-06-10 01:15:39

您可以使用tf.expand_dims(a,0)和tf.expand_dims(b,1)来获得等级2的形状。尝试以下代码:

代码语言:javascript
复制
a = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], name='a')
b = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], name='b')
c = tf.matmul(tf.expand_dims(a,0), tf.expand_dims(b, 1))
c2=tf.squeeze(c)
sess=tf.Session()
print(sess.run(c))
print(sess.run(c2))enter code here

它将显示:

代码语言:javascript
复制
[[ 91.]]
91.0
票数 0
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/54703832

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档