首页 > 其他 > 详细

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

时间:2019-02-03 15:42:18      阅读:181      评论:0      收藏:0      [点我收藏+]

译者:cangyunye

作者: Adam Paszke

修订者: Adam Dziedzic

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

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

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

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

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

import torch
from torch.autograd import Function

无参数示例

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

它只是被恰当的命名为BadFFTFunction

本层的实现方式

from numpy.fft import rfft2, irfft2

阅读全文/改进本文

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

原文:https://www.cnblogs.com/wizardforcel/p/10350457.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!