cafe 的深度学习中 var.diff[...] = 0 是什么意思?
what means var.diff[...] = 0 in deeplearning by caffe?
这里是caffe cnn中使用backward的函数,
顶部是这一层的输出,底部是前一层的输入
def backward(self,top,propagate_down,bottom):
if propagate_down[0] and self.count!=0:
bottom[0].diff[...]=0
bottom[0].diff[self.valid_index]=top[0].diff[...]
if propagate_down[1] and self.count!=0:
bottom[1].diff[...]=0
bottom[1].diff[self.valid_index]=top[1].diff[...]
呃....diff[...]=0 是什么意思?
设置
diff[...]=0
简单地将var
的所有个梯度设置为零。
根据 , you can read more about ellipsis ([...]
) here 的建议。
这里是caffe cnn中使用backward的函数, 顶部是这一层的输出,底部是前一层的输入
def backward(self,top,propagate_down,bottom):
if propagate_down[0] and self.count!=0:
bottom[0].diff[...]=0
bottom[0].diff[self.valid_index]=top[0].diff[...]
if propagate_down[1] and self.count!=0:
bottom[1].diff[...]=0
bottom[1].diff[self.valid_index]=top[1].diff[...]
呃....diff[...]=0 是什么意思?
设置
diff[...]=0
简单地将var
的所有个梯度设置为零。
根据 [...]
) here 的建议。