xiaobocasia / sv-x-softmax Goto Github PK

new_zy = mx.sym.where(cond, new_zy, zy_keep)#>0 s*cos(@+m), <0 cut
diff = new_zy - zy#为了将真实的标签位置替换为new_zy
diff = mx.sym.expand_dims(diff, 1)

new_zy = mx.sym.expand_dims(new_zy, 1)
gt_one_hot_down = mx.sym.one_hot(gt_label, depth = args.num_classes, on_value = 0.0, off_value = 1.0)
gt_zy = mx.sym.broadcast_mul(gt_one_hot_down,fc7)#求非gt的Cos值

gt_greater = mx.sym.broadcast_greater(gt_zy,new_zy)#得到非gt值比gt值大的标签索引
gt_lesser_than = mx.sym.broadcast_lesser_equal(gt_zy,new_zy)#得到非gt值小或等于gt值的标签索引
gt_greater_mul = gt_greater*t#非gt位置*t
gt_greater_mul = mx.sym.broadcast_add(gt_greater_mul,gt_lesser_than)#再加上gt_lesser_than,为了与gt_zy相乘
fc7 = mx.sym.broadcast_mul(fc7,gt_greater_mul)#对应位置乘t
gt_greater_add = gt_greater*(t-1)#对应位置变为t-1
fc7 = mx.sym.broadcast_add(fc7,gt_greater_add)#对应位置加上t-1   

gt_one_hot = mx.sym.one_hot(gt_label, depth = args.num_classes, on_value = 1.0, off_value = 0.0)
body = mx.sym.broadcast_mul(gt_one_hot, diff)
fc7 = fc7+body