如何解决Tensorfow.js找到张量的中位数
我需要找到角度应用中通过tensorflow.js读取为张量的图像的中值。谁能建议如何在tfjs中找到中位数?
像-
let x = tf.tensor([1,2,3,4,5,8,9]);
console.log(x.median());
这应该打印4。
我试图在javascript中找到等效的张量流概率统计信息,但到目前为止还没有运气。
解决方法
中位数是排列序列后的中间数字。张量需要首先排序。索引sizeTensor / 2
上的值为中间值。
t= tf.tensor([1,2,3,4,5,8,9]);
tf.topk(t,t.size).values.slice(t.size / 2,1).print() // will print 4
要使以上代码与高维张量配合使用(例如图像),我们首先需要将其整形为一维张量
给出img
图像的张量
t = img.reshape([-1])
// do the same as above
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。