tensorflow 去重

来源:互联网 发布:数控车床计算软件下载 编辑:程序博客网 时间:2024/06/16 00:47
import tensorflow as tfa = tf.constant([1,2,2,3,3])b = tf.unique(a)sess = tf.Session()print(sess.run(b))

print结果
Unique(y=array([1, 2, 3]), idx=array([0, 1, 1, 2, 2]))

原创粉丝点击