提交 7691524a authored 作者: Iban Harlouchet's avatar Iban Harlouchet

numpydoc for theano/sandbox/cuda/GpuConv3D.py

上级 2d9e40e3
...@@ -12,7 +12,10 @@ from theano.sandbox.cuda import CudaNdarrayType, GpuOp ...@@ -12,7 +12,10 @@ from theano.sandbox.cuda import CudaNdarrayType, GpuOp
class GpuConv3D(GpuOp): class GpuConv3D(GpuOp):
""" GPU implementation of Conv3D """ """
GPU implementation of Conv3D.
"""
def __eq__(self, other): def __eq__(self, other):
return type(self) == type(other) return type(self) == type(other)
...@@ -25,10 +28,18 @@ class GpuConv3D(GpuOp): ...@@ -25,10 +28,18 @@ class GpuConv3D(GpuOp):
def make_node(self, V, W, b, d): def make_node(self, V, W, b, d):
""" """
:param V: Visible unit, input
:param W: Weights, filter Parameters
:param b: bias ----------
:param d: strides when moving the filter over the input V
Visible unit, input.
W
Weights, filter.
b
Bias.
d
Strides when moving the filter over the input.
""" """
V_ = as_cuda_ndarray_variable(V) V_ = as_cuda_ndarray_variable(V)
W_ = as_cuda_ndarray_variable(W) W_ = as_cuda_ndarray_variable(W)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论