keras (tensorflow backend) conditional assignment with K.switch()
问题 I'm trying to implement something like if np.max(subgrid) == np.min(subgrid): middle_middle = cur_subgrid + 1 else: middle_middle = cur_subgrid Since the condition can only be determined at run-time, I'm using Keras syntax as following middle_middle = K.switch(K.max(subgrid) == K.min(subgrid), lambda: tf.add(cur_subgrid,1), lambda: cur_subgrid) But I'm getting this error: <ipython-input-112-0504ce070e71> in col_loop(j, gray_map, mask_A) 56 57 ---> 58 middle_middle = K.switch(K.max(subgrid) ==