Tensorflow报错: Shape must be rank 2 but is rank 1 for 'MatMul' (op: 'MatMul') with input shapes

本文阅读 1 分钟
首页 代码,Java 正文
import tensorflow as tf

# 创建两个常量Tensor
const1 = tf.constant([2, 2])
const2 = tf.constant([4, 4])

multiple = tf.matmul(const1, const2)

# 尝试用print输出multiple的值
print(multiple)

初学TF时矩阵乘法报错:

ValueError: Shape must be rank 2 but is rank 1 for 'MatMul' (op: 'MatMul') with input shapes: [2], [2].

 

解决方法:

矩阵相乘是对应的行和列相乘的结果,所以需要第一个矩阵的列数与第二个矩阵的行数相等。

只需令

const1 = tf.constant([[2, 2]])
const2 = tf.constant([[4],[4]])

 

本文为互联网自动采集或经作者授权后发布,本文观点不代表立场,若侵权下架请联系我们删帖处理!文章出自:https://wangjiawei.blog.csdn.net/article/details/89152559
-- 展开阅读全文 --
安全面试之XSS(跨站脚本攻击)
« 上一篇 07-24

发表评论

成为第一个评论的人

热门文章

标签TAG

最近回复