pytorch 1 7

1conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

1conda install pytorch torchvision torchaudio cudatoolkit=11.0 -c pytorch

1>>> import torch.fft
2>>> t = torch.arange(4)
3>>> t
4tensor([0, 1, 2, 3])
5
6>>> torch.fft.fft(t)
7tensor([ 6.+0.j, -2.+2.j, -2.+0.j, -2.-2.j])
8
9>>> t = tensor([0.+1.j, 2.+3.j, 4.+5.j, 6.+7.j])
10>>> torch.fft.fft(t)
11tensor([12.+16.j, -8.+0.j, -4.-4.j,  0.-8.j])
12