@@ -418,8 +418,8 @@ def solve_semi_dual_entropic(a, b, M, reg, method, numItermax=10000, lr=None,
418
418
return None
419
419
420
420
opt_alpha = c_transform_entropic (b , M , reg , opt_beta )
421
- pi = (np .exp ((opt_alpha [:, None ] + opt_beta [None , :] - M [:, :]) / reg )
422
- * a [:, None ] * b [None , :])
421
+ pi = (np .exp ((opt_alpha [:, None ] + opt_beta [None , :] - M [:, :]) / reg ) *
422
+ a [:, None ] * b [None , :])
423
423
424
424
if log :
425
425
log = {}
@@ -520,8 +520,8 @@ def batch_grad_dual(a, b, M, reg, alpha, beta, batch_size, batch_alpha,
520
520
arXiv preprint arxiv:1711.02283.
521
521
'''
522
522
523
- G = - (np .exp ((alpha [batch_alpha , None ] + beta [None , batch_beta ]
524
- - M [batch_alpha , :][:, batch_beta ]) / reg ) *
523
+ G = - (np .exp ((alpha [batch_alpha , None ] + beta [None , batch_beta ] -
524
+ M [batch_alpha , :][:, batch_beta ]) / reg ) *
525
525
a [batch_alpha , None ] * b [None , batch_beta ])
526
526
grad_beta = np .zeros (np .shape (M )[1 ])
527
527
grad_alpha = np .zeros (np .shape (M )[0 ])
@@ -702,8 +702,8 @@ def solve_dual_entropic(a, b, M, reg, batch_size, numItermax=10000, lr=1,
702
702
703
703
opt_alpha , opt_beta = sgd_entropic_regularization (a , b , M , reg , batch_size ,
704
704
numItermax , lr )
705
- pi = (np .exp ((opt_alpha [:, None ] + opt_beta [None , :] - M [:, :]) / reg )
706
- * a [:, None ] * b [None , :])
705
+ pi = (np .exp ((opt_alpha [:, None ] + opt_beta [None , :] - M [:, :]) / reg ) *
706
+ a [:, None ] * b [None , :])
707
707
if log :
708
708
log = {}
709
709
log ['alpha' ] = opt_alpha
0 commit comments