提交 9d7d952c authored 作者: Frederic's avatar Frederic

Better doc and add missing import

上级 8e7a76a2
...@@ -19,3 +19,4 @@ and ops which are particular to neural networks and deep learning. ...@@ -19,3 +19,4 @@ and ops which are particular to neural networks and deep learning.
conv conv
nnet nnet
neighbours neighbours
bn
...@@ -6,3 +6,4 @@ from .ConvTransp3D import * ...@@ -6,3 +6,4 @@ from .ConvTransp3D import *
from .sigm import (softplus, sigmoid, sigmoid_inplace, from .sigm import (softplus, sigmoid, sigmoid_inplace,
scalar_sigmoid, ultra_fast_sigmoid, scalar_sigmoid, ultra_fast_sigmoid,
hard_sigmoid) hard_sigmoid)
from .bn import batch_normalization
...@@ -35,6 +35,8 @@ def batch_normalization(inputs, gamma, beta, mean, std, ...@@ -35,6 +35,8 @@ def batch_normalization(inputs, gamma, beta, mean, std,
to a set of activations. to a set of activations.
Work also on GPU Work also on GPU
.. versionadded:: 0.7.1
Parameters Parameters
---------- ----------
inputs : symbolic tensor inputs : symbolic tensor
......
...@@ -2002,6 +2002,8 @@ def relu(x, alpha=0): ...@@ -2002,6 +2002,8 @@ def relu(x, alpha=0):
""" """
Compute the element-wise rectified linear activation function. Compute the element-wise rectified linear activation function.
.. versionadded:: 0.7.1
Parameters Parameters
---------- ----------
x : symbolic tensor x : symbolic tensor
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论