AMLab-Amsterdam / lie_learn

Computations involving Lie groups and harmonic analysis
MIT License
190 stars 47 forks source link

TypeError: unsupported operand type(s) for *: 'numpy.ndarray' and 'Tensor' #18

Closed WN1695173791 closed 1 month ago

WN1695173791 commented 3 years ago

Traceback (most recent call last): File "/home/wn/视频/se3-transformer-public-master2/equivariant_attention/from_se3cnn/SO3.py", line 284, in _test_change_basis_wigner_to_rot() File "/home/wn/视频/se3-transformer-public-master2/equivariant_attention/from_se3cnn/SO3.py", line 264, in _test_change_basis_wigner_to_rot print(wigner_D_matrix(1, a, b, c)) File "/home/wn/anaconda3/envs/so3/lib/python3.6/site-packages/lie_learn/representations/SO3/wigner_d.py", line 63, in wigner_D_matrix D = rot_mat(alpha=alpha, beta=beta, gamma=gamma, l=l, J=Jd[l]) File "/home/wn/anaconda3/envs/so3/lib/python3.6/site-packages/lie_learn/representations/SO3/pinchon_hoggan/pinchon_hoggan_dense.py", line 79, in rot_mat Xa = z_rot_mat(alpha, l) File "/home/wn/anaconda3/envs/so3/lib/python3.6/site-packages/lie_learn/representations/SO3/pinchon_hoggan/pinchon_hoggan_dense.py", line 62, in z_rot_mat M[inds, reversed_inds] = np.sin(frequencies angle) TypeError: unsupported operand type(s) for : 'numpy.ndarray' and 'Tensor'

I had this problem,and is there the same problem for anyone?