33 lines
432 B
Python
33 lines
432 B
Python
# -*- encoding:utf-8 -*-
|
|
|
|
'''
|
|
@Author : dingjiawen
|
|
@Date : 2023/11/15 13:18
|
|
@Usage :
|
|
@Desc :
|
|
'''
|
|
|
|
import torch_dct as dct
|
|
import numpy as np
|
|
import torch
|
|
|
|
a = np.load("a.npy")
|
|
b = np.load("b.npy")
|
|
|
|
a = torch.tensor(a)
|
|
|
|
c = dct.dct(x=a)
|
|
|
|
d = []
|
|
|
|
for i in range(a.shape[1]):
|
|
d.append(dct.dct(a[:,i,:]))
|
|
d = torch.stack(d,dim=1)
|
|
|
|
torch.nn.Linear()
|
|
print("a.shape:",a.shape)
|
|
print("a:",a)
|
|
print("b:",b)
|
|
print("c:",c)
|
|
print("d:",d)
|