Тензор потока: есть ли способ узнать, сколько времени занимает операция прямой связи?
У меня есть модель, где я выполняю l1 нормализацию по весам. Теперь у меня есть две операции: одна для нахождения l1-нормы весов. Другая операция умножает эти веса на их l1-норму. Первая операция должна быть тривиальной в случае логического вывода (поскольку веса будут известны). Тем не менее, я хотел бы убедиться, какая операция является виновником времени.
Вот как я нахожу l1-норму:
alpha = tf.reduce_sum(tf.multiply(w_conv, tf.sign(w_conv)), axis=[0,1,2], keep_dims=False)/alpha_constant
Вот как я нормализую:
conv = tf.conv2d(data, alpha*w_conv, strides=[1, 1, 1, 1], padding='SAME')
Так как же определить, сколько времени занимает каждая из этих операций в TensorFlow?