Commit 08e3e5cf authored by YIN Zihua's avatar YIN Zihua
Browse files

@zihua: finished the translation to Python3

parent 32f64dbe
......@@ -103,10 +103,16 @@
%% Cell type:code id: tags:
``` python
%% Cell type:code id: tags:
``` python
# gram matrix and loss
class GramMatrix(nn.Module):
def forward(self, input):
b,c,h,w = input.size()
F = input.view(b, c, h*w)
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment