|
|
@ -29,9 +29,6 @@ def spectral_norm(weight, u, v, dim, power_iters, eps):
|
|
|
|
if dim != 0:
|
|
|
|
if dim != 0:
|
|
|
|
perm = [dim] + [d for d in range(len(shape)) if d != dim]
|
|
|
|
perm = [dim] + [d for d in range(len(shape)) if d != dim]
|
|
|
|
weight_mat = weight_mat.transpose(perm)
|
|
|
|
weight_mat = weight_mat.transpose(perm)
|
|
|
|
real_shape = weight_mat.shape
|
|
|
|
|
|
|
|
else:
|
|
|
|
|
|
|
|
real_shape = shape
|
|
|
|
|
|
|
|
weight_mat = weight_mat.reshape((h, w))
|
|
|
|
weight_mat = weight_mat.reshape((h, w))
|
|
|
|
|
|
|
|
|
|
|
|
u = u.reshape((h, 1))
|
|
|
|
u = u.reshape((h, 1))
|
|
|
|