tensor操作
tensor比大
比大的实现,当tensor中元素x大于等于5时,对应输出tensor中元素y=x * 2;否则 y= x * 3的计算
点击
同理可用 tf.less 替换 tf.greater
tensor数据提取
假设提取 小于1 的损失值
loss为 计算出来的损失
loss_one = tf.zeros_like( loss ) +1.0
rpn_select = tf.where ( tf.less( loss , loss_one) )
loss = tf.gather(loss, rpn_select)
训练时报错:ResourceExhaustedError
强制杀掉进程:
nvidia-smi
sudo kill -9 进程号