julius/prototorch
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

feat: add repr for LinearTransform

Browse Source
This commit is contained in:
Jensun Ravichandran 2022-04-01 10:13:25 +02:00
parent 46ff1c4eb1
commit 85f75bb28c
No known key found for this signature in database
GPG Key ID: 4E9348239810B51F
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
prototorch/core/transforms.py
View File

@ -38,6 +38,9 @@ class LinearTransform(torch.nn.Module):
def forward(self, x): def forward(self, x):
return x @ self._weights return x @ self._weights
def extra_repr(self):
return f"weights: (shape: {tuple(self._weights.shape)})"
# Aliases # Aliases
Omega = LinearTransform Omega = LinearTransform