#NEWLINE# def test_fftshift_axes( ) : #NEWLINE# #INDENT# def fun( x ) : #NEWLINE# #INDENT# return to_scalar ( np . fft . fftshift ( x , axes = 1 ) ) #NEWLINE# #UNINDENT# d_fun = ( lambda x : to_scalar ( grad ( fun ) ( x ) ) ) #NEWLINE# D = 5 #NEWLINE# mat = ( npr . randn ( d_fun , D ) / 10.0 ) #NEWLINE# check_grads ( fun , mat ) #NEWLINE# check_grads ( d_fun , mat )