代码之家 › 专栏 › 技术社区 › JM Espez

TensorFlow:tf.matrix_inverse()中的形状秩2

matrix-inverse tensorflow

0

JM Espez · 技术社区 · 9 年前

我尝试tf时遇到了问题。matrix_inverse()方法。

我有一个维度为[17400400]的张量。我们可以阅读API文档:

输入是形状为[…,M,M]的张量,其最内部的2维构成方阵。

在我的情况下,M=400。当我尝试计算17逆矩阵时:

Out[56]: L
Out[57]: <tf.Tensor 'while_4/Exit_1:0' shape=(17, 400, 400) dtype=float32>
Out[59]: InvL = tf.matrix_inverse(L)
        ValueError: Shape (17, 400, 400) must have rank 2

我知道我有一个形状等级2,但这个方法失败了。我不知道我是否做错了什么。需要帮忙吗?

1 回复 | 直到 9 年前

1

Zhongyu Kuang 9 年前

您参考的API文档仅适用于tensorflow版本r0.11或主版本。

import tensorflow
tensorflow.__version__

对于r0.11之前的版本,如 documentation 这里说:

它不支持使用秩反转张量=2.

尝试 update the tensorflow to r0.11 或者直接从github主分支提取