У меня есть тензор измерения [n_samples, height, width], т. Е. Много изображений.
У меня есть другой тензор измерения [n_samples,], т. Е. Каждый элемент этого тензора является индексом кластера одного изображения в первом тензоре.
Теперь, как я могу вычислить дисперсиюкаждый кластер (кластеризация первого тензора основана на втором тензоре) в Tensorflow?
Я нашел некоторую функцию Tensorflow, например unsorted_segment_mean, которая выполняет аналогичные действия, но я не могу понять, как рассчитать дисперсию?Кто-нибудь может мне помочь с этим?Спасибо!