Skip to content

Commit ec47c8a

Browse files
authored
Update layer.py
1 parent d9d4235 commit ec47c8a

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

layer.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@ def __init__(self):
1111
super(nconv,self).__init__()
1212

1313
def forward(self,x, A):
14-
x = torch.einsum('ncvl,vw->ncwl',(x,A))
14+
x = torch.einsum('ncwl,vw->ncvl',(x,A))
1515
return x.contiguous()
1616

1717
class dy_nconv(nn.Module):
@@ -325,4 +325,4 @@ def forward(self, input, idx):
325325

326326
def extra_repr(self):
327327
return '{normalized_shape}, eps={eps}, ' \
328-
'elementwise_affine={elementwise_affine}'.format(**self.__dict__)
328+
'elementwise_affine={elementwise_affine}'.format(**self.__dict__)

0 commit comments

Comments
 (0)