-
Notifications
You must be signed in to change notification settings - Fork 282
Open
Description
In
Line 232 in c750fec
x_l = torch.sum(x_0 * x_l, 1).view([-1,1]) * getattr(self,'cross_weight_'+str(i+1)).view([1,-1]) + getattr(self,'cross_bias_'+str(i+1)) + x_l |
x_0 * x_l
should be replaced by torch.matmul(x_0, x_l.t())
, right ?Metadata
Metadata
Assignees
Labels
No labels