input must have last dimension >= k = 3 but is 2 for 'TopKV2_这种错误是怎么产生的

import tensorflow as tf import numpy as np # 创建一个长度为10的向量,内部数据随机生成。 a = tf.convert_to_tensor(np.random.random([10,2])) # 取出前5个最大的数据,默认从大到小进行排序。 b = tf.nn.top_k(a, 3) with tf.Session() as sess:
相关文章
相关标签/搜索