tmp=tf.gather(y_true_f,[1,3],axis=1) y_core=K.sum(tmp,axis=1)解释
时间: 2023-11-01 16:08:16 浏览: 53
TMP算法
5星 · 资源好评率100%
这段代码使用了TensorFlow中的gather函数,从y_true_f中按照索引[1,3]提取特定的列(即axis=1方向上的元素),并将提取出来的结果存储到tmp中。接着使用K.sum函数将tmp中的所有元素沿axis=1方向求和,得到一个一维向量y_core。换句话说,y_core是y_true_f中第1列和第3列元素之和的结果。
阅读全文