gpt4 book ai didi

python - 值错误 : Shape must be rank 2 but is rank 3 for 'MatMul'

转载 作者:行者123 更新时间:2023-11-28 18:12:50 26 4
gpt4 key购买 nike

我有以下 TensorFlow 代码:

layer_1 = tf.add(tf.matmul(tf.cast(x, tf.float32), weights['h1']), biases['b1'])

但是抛出以下错误:

ValueError: Shape must be rank 2 but is rank 3 for 'MatMul' (op: 'MatMul') with input shapes: [?,5741,20000], [20000,128].

它表示 x 的形状为 (?,5741,20000)。如何将 x 的形状转换为 (5741, 20000)?

提前致谢!

最佳答案

我建议使用张量点积而不是简单的矩阵乘法来保持批量大小。这个答案比 @mrry 更笼统。

layer_1 = tf.add(tf.tensordot(tf.cast(x, tf.float32), weights['h1'], [[2], [0]]), biases['b1'])

关于python - 值错误 : Shape must be rank 2 but is rank 3 for 'MatMul' ,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/50162787/

26 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com