refine test_spectral_norm. test=develop

revert-16045-imperative_remove_desc
dengkaipeng 6 years ago committed by ceci3
parent 12416a24d2
commit 91f8531586

@ -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))

Loading…
Cancel
Save