Я конвертирую numpy в тензорный поток. Входные данные представляют собой изображение размером 15x15 и 470 входов. Ниже numpy выполняет корреляцию для каждого входного изображения. Но я хочу написать код тензорного потока, чтобы выполнить корреляцию для всех изображений 470 за один раз. Не могли бы вы дать мне знать, как конвертировать ниже в tenorflow?
# numpy
AlarmsImage = ReadImage(i) # i ranges 0 to 469. AlarmsImage 15x15
AlarmsIndex = np.asarray(np.where(AlarmsImage))
CentralAlarms = AlarmsIndex - 7
corr = np.matmul(CentralAlarms, CentralAlarms.transpose()) / CentralAlarms.shape[1]
# tensorflow
AlarmsImage3D = ReadAllImage() # 470x15x15
AlarmsIndex = tf.where(AlarmsImage3D)
# No idea how to proceed
# I might need to use tfp.stats.correlation or tensordot ??