tf.math.equal() 函数讲解

def full_similarity_loss(y_true, y_pred): # 增强数值稳定性 EPSILON = 1e-5 # 改进哈希码生成 relaxed_hash = tf.tanh(TEMPERATURE * y_pred) normalized_hash = tf.math.l2_normalize(relaxed_hash, axis=1) # 安全相似度计算 sim_matrix = tf.matmul(normalized_hash, normalized_hash, transpose_b=True) sim_matrix = tf.clip_by_value(sim_matrix, -0.999, 0.999) # 严格限制范围 # 构建真实相似矩阵（添加调试断言） S = tf.cast(tf.equal(y_true[:, None], y_true[None, :]), tf.float32) tf.debugging.assert_non_negative(S) # 验证标签矩阵 # 改进损失计算 safe_pos = (1 - sim_matrix) / ALPHA pos_term = tf.math.log(1 + tf.nn.relu(safe_pos) + EPSILON) # 防止负数 safe_neg = sim_matrix / ALPHA neg_term = tf.math.log(1 + tf.nn.relu(safe_neg) + EPSILON) # 添加正则化项 hash_std = tf.math.reduce_std(y_pred) regularization = 0.1 * tf.square(hash_std - 0.5) return (tf.reduce_mean(pos_term) + tf.reduce_mean(neg_term)) * 0.5 + regularization

def evaluate(inp_SNR,noise,eq_flag,epsilon=1e-12): s_logits = logit_model(inp_SNR) s = s_model(s_logits) soft_bits = soft_bit_encoder(s) hard_bits = hard_decision_on_bit(soft_bits) enc = Trans_model_bit(hard_bits) bit_set = tf.math.mod(tf.bitwise.right_shift(tf.expand_dims(symbol_set,1), tf.range(bitlen)), 2) bit_set = tf.reverse(bit_set, axis = [-1]) constellation = Trans_model_bit(bit_set) constellation = constellation - tf.reduce_mean(constellation) constellation = tf.expand_dims(constellation, 0) p_s = tf.nn.softmax(s_logits) norm_factor= tf.math.rsqrt(tf.maximum(p_norm(p_s, constellation),epsilon)) norm_constellation = r2c(norm_factor) * constellation should_always_be_one = p_norm(p_s, norm_constellation) x = r2c(norm_factor) * enc Tx = upsample_pulse_shaping(x, Fs, PS_filter) Rx = Tx + noise y = Model_Eq(Rx) entropy_S = -p_norm(p_s, p_s, lambda x: log2(x)) GMI = GMIcal_tf(x,tf.squeeze(y),M,norm_constellation,hard_bits_out,p_s) NGMI = 1-(entropy_S-GMI)/bitlen loss_NGMI = tf.nn.relu(NGMI_th-NGMI) loss_Eq = tf.reduce_mean(tf.square(tf.abs(x-y))) loss = loss_Eq*eq_flag*0.5-GMI+loss_NGMI*100 return loss.numpy(), loss_Eq.numpy(), p_s.numpy(), GMI.numpy(), NGMI.numpy(), entropy_S, norm_constellation.numpy(), x.numpy(), y.numpy() 继续，修改这部分代码