File size: 241 Bytes
e9fe176 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
require 'GDLCriterion'
a = torch.ones(2, 2, 2, 2)
b = torch.zeros(2, 2, 2, 2)
c = nn.GDLCriterion(2)
l = c:forward(a, b)
d = c:backward(a, b)
print(l)
print(d)
b[1][1][1][1] = 1
l = c:forward(a, b)
d = c:backward(a, b)
print(l)
print(d)
|