问题描述
我正在尝试连接 2 个float64类型的张量。
tf.concat(predictions[:,:5], max_ind)
但是我收到一个错误
ValueError: Tensor conversion requested dtype int32 for Tensor with dtype float64: <tf.Tensor: id=44, shape=(4800,), dtype=float64, numpy=array([0., 0., 0., ..., 0., 0., 0.])>
我不明白，因为张量都是float64 。
1楼
tf.concat values连接到应用操作的axis 。
据我从错误中了解到，模型认为max_ind提供的max_ind是axis ，这就是为什么它给您一个错误，说轴不应该是浮点数，而是整数。
如果您尝试连接predictions[:,:5]和max_ind ，则应按如下方式使用它：
tf.concat([predictions[:,:5], max_ind], -1)
我使用 -1 作为轴，但您可以根据需要调整它。