gpt4 book ai didi

tensorflow 值错误 : Shape must be rank 1 but is rank 2

转载 作者:行者123 更新时间:2023-12-04 02:08:04 29 4
gpt4 key购买 nike

import tensorflow as tf
x = [[1,2,3],[4,5,6]]
y = [0,1]
z = [1,2]
x = tf.constant(x)
y = tf.constant(y)
z = tf.constant(z)
m = x[y,z]

我期望的是 m = [2,6]
我可以通过 theano 或 numpy 得到结果。我如何使用 tensorflow 获得结果?

最佳答案

你会想要使用 tf.gather_nd

   slices = tf.gather_nd(x, [y, z])

希望这可以帮助。

关于 tensorflow 值错误 : Shape must be rank 1 but is rank 2,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/40366163/

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