PyTorch 1.0 中文官方教程：用 numpy 和 scipy 创建扩展

译者：cangyunye

作者: Adam Paszke

修订者: Adam Dziedzic

在这个教程里，我们要完成两个任务:

1. 创建一个无参神经网络层。

   这里需要调用numpy作为实现的一部分。

2. 创建一个权重自主优化的伸进网络层。

   这里需要调用Scipy作为实现的一部分。

import torch
from torch.autograd import Function

无参数示例

这一层并没有特意做什么任何有用的事或者去进行数学上的修正。

它只是被恰当的命名为BadFFTFunction

本层的实现方式

from numpy.fft import rfft2, irfft2

阅读全文／改进本文

转载于:https://www.cnblogs.com/wizardforcel/p/10350457.html