basis = radial_window(*_sample_cube(size, order_in, order_out))
if basis is not None:
// normalize filter energy (not over axis 0, i.e. different filters are normalized independently)
basis = basis / np.sqrt(np.sum(basis ** 2, axis=(1, 2, 3, 4, 5), keepdims=True))
return basis
////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
After Change
basis = radial_window(*_sample_cube(size, order_in, order_out))
if basis is not None:
// normalize filter energy (not over axis 0, i.e. different filters are normalized independently)
basis = basis / basis.view(basis.size(0), -1).norm(dim=1).view(-1, 1, 1, 1, 1, 1)
return basis
////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////