Tensorfow.js tìm trung bình của tensor

Question

Tôi cần tìm trung vị của một hình ảnh được đọc dưới dạng tensor thông qua tensorflow.js trong ứng dụng góc. Bất cứ ai có thể đề nghị làm thế nào để tìm trung vị trong tfjs?

Giống -

let x = tf.tensor([1, 2, 3, 4, 5, 8, 9]);
console.log(x.median());

Điều này sẽ in 4.

Tôi đã thử tìm số liệu thống kê xác suất tensorflow tương đương trong javascript nhưng không có may mắn cho đến nay.

Answer 1

Số trung vị là số ở giữa sau khi dãy số đã được sắp xếp. Đầu tiên, tensor cần được sắp xếp. Giá trị tại chỉ mục sizeTensor / 2sẽ là giá trị trung vị.

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

Để ở trên hoạt động với tensor kích thước cao - ví dụ như hình ảnh, trước tiên chúng ta sẽ cần định hình lại tensor 1d

Cho imgcăng thẳng của một hình ảnh

t = img.reshape([-1])
// do the same as above