2020年10月15日
循环相关 pytorch
摘要： # pytorch 1.6 目前pytorch.fft没有支持复数数据类型，如果支持就好写很多了 # 当onesided为True时，依据输入张量的嵌入维度奇偶性，可能出现维度不匹配的问题 from einops impor rearrange def ccorr(a, b, sigdim=1, o 阅读全文
posted @ 2020-10-15 14:43 e-yi 阅读(19) 评论(0) 推荐(0)