Skip to content
项目
群组
代码片段
帮助
当前项目
正在载入...
登录 / 注册
切换导航面板
P
pytensor
项目
项目
详情
活动
周期分析
仓库
仓库
文件
提交
分支
标签
贡献者
图表
比较
统计图
议题
0
议题
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
CI / CD
CI / CD
流水线
作业
日程
统计图
Wiki
Wiki
代码片段
代码片段
成员
成员
折叠边栏
关闭边栏
活动
图像
聊天
创建新问题
作业
提交
问题看板
Open sidebar
testgroup
pytensor
Commits
d3ca3329
提交
d3ca3329
authored
2月 24, 2014
作者:
abergeron
浏览文件
操作
浏览文件
下载
差异文件
Merge pull request #1747 from nouiz/doc
[MRG]Doc
上级
ab720fb5
0b4c5463
全部展开
显示空白字符变更
内嵌
并排
正在显示
14 个修改的文件
包含
152 行增加
和
103 行删除
+152
-103
index.txt
doc/library/sandbox/index.txt
+1
-3
rng_mrg.txt
doc/library/sandbox/rng_mrg.txt
+16
-0
index.txt
doc/library/sparse/index.txt
+44
-40
raw_random.txt
doc/library/tensor/raw_random.txt
+0
-0
examples.txt
doc/tutorial/examples.txt
+16
-11
multi_cores.txt
doc/tutorial/multi_cores.txt
+30
-17
check_blas.py
theano/misc/check_blas.py
+1
-0
test_type.py
theano/sandbox/cuda/tests/test_type.py
+6
-3
rng_mrg.py
theano/sandbox/rng_mrg.py
+2
-0
basic.py
theano/scalar/basic.py
+2
-1
basic.py
theano/sparse/basic.py
+13
-8
Conv3D.py
theano/tensor/nnet/Conv3D.py
+7
-3
conv3d2d.py
theano/tensor/nnet/conv3d2d.py
+4
-0
raw_random.py
theano/tensor/raw_random.py
+10
-17
没有找到文件。
doc/library/sandbox/index.txt
浏览文件 @
d3ca3329
...
@@ -16,6 +16,4 @@
...
@@ -16,6 +16,4 @@
cuda/index
cuda/index
linalg
linalg
neighbours
neighbours
rng_mrg
doc/library/sandbox/rng_mrg.txt
0 → 100644
浏览文件 @
d3ca3329
.. _libdoc_rng_mrg:
===================================================================
:mod:`sandbox.rng_mrg` -- MRG random number generator
===================================================================
.. module:: sandbox.rng_mrg
:platform: Unix, Windows
:synopsis: MRG random number generator
.. moduleauthor:: LISA
API
===
.. automodule:: theano.sandbox.rng_mrg
:members:
doc/library/sparse/index.txt
浏览文件 @
d3ca3329
...
@@ -11,20 +11,20 @@ In the tutorial section, you can find a :ref:`sparse tutorial
...
@@ -11,20 +11,20 @@ In the tutorial section, you can find a :ref:`sparse tutorial
The sparse submodule is not loaded when we import Theano. You must
The sparse submodule is not loaded when we import Theano. You must
import ``theano.sparse`` to enable it.
import ``theano.sparse`` to enable it.
The sparse module provide
the same functionalities
as the tensor
The sparse module provide
s the same functionality
as the tensor
module. The difference lies under the cover because sparse matrices
module. The difference lies under the cover
s
because sparse matrices
do
es
not store data in a contiguous array. Note that there are no GPU
do not store data in a contiguous array. Note that there are no GPU
implementations for sparse matrices i
mplemented in Theano. The sparse
implementations for sparse matrices i
n Theano. The sparse module has
module has
been used in:
been used in:
- NLP: Dense linear transformations of sparse vectors.
- NLP: Dense linear transformations of sparse vectors.
- Audio: Filterbank in Fourier domain.
- Audio: Filterbank in
the
Fourier domain.
Compressed Sparse Format
Compressed Sparse Format
========================
========================
This section tries to explain how information is store for the two
This section tries to explain how information is store
d
for the two
sparse formats of SciPy supported by Theano. There
is
more formats
sparse formats of SciPy supported by Theano. There
are
more formats
that can be used with SciPy and some documentation about them may be
that can be used with SciPy and some documentation about them may be
found `here
found `here
<http://deeplearning.net/software/theano/sandbox/sparse.html>`_.
<http://deeplearning.net/software/theano/sandbox/sparse.html>`_.
...
@@ -50,14 +50,14 @@ attributes: ``data``, ``indices``, ``indptr`` and ``shape``.
...
@@ -50,14 +50,14 @@ attributes: ``data``, ``indices``, ``indptr`` and ``shape``.
CSC Matrix
CSC Matrix
----------
----------
In the *Compressed Sparse Column* format, ``indices`` stands for
index
In the *Compressed Sparse Column* format, ``indices`` stands for
in
side the column vectors of the matrix and ``indptr`` tells where the
in
dexes inside the column vectors of the matrix and ``indptr`` tells
column starts in the ``data`` and in the ``indices``
where the
column starts in the ``data`` and in the ``indices``
attributes. ``indptr`` can be t
ought as giving the slice which must be
attributes. ``indptr`` can be t
hought of as giving the slice which
applied to the other attribute in order to get each column of the
must be applied to the other attribute in order to get each column of
matrix. In other words, ``slice(indptr[i], indptr[i+1])`` correspond
the matrix. In other words, ``slice(indptr[i], indptr[i+1])``
to the slice needed to find the i-th column of the matrix in the
corresponds to the slice needed to find the i-th column of the matrix
``data`` and in the
``indices`` fields.
in the ``data`` and
``indices`` fields.
The following example builds a matrix and returns its columns. It
The following example builds a matrix and returns its columns. It
prints the i-th column, i.e. a list of indices in the column and their
prints the i-th column, i.e. a list of indices in the column and their
...
@@ -84,18 +84,18 @@ corresponding value in the second list.
...
@@ -84,18 +84,18 @@ corresponding value in the second list.
CSR Matrix
CSR Matrix
----------
----------
In the *Compressed Sparse Row* format, ``indices`` stands for index
In the *Compressed Sparse Row* format, ``indices`` stands for index
es
inside the row vectors of the matrix and ``indptr`` tells where the
inside the row vectors of the matrix and ``indptr`` tells where the
row starts in the ``data`` and in the ``indices``
row starts in the ``data`` and in the ``indices``
attributes. ``indptr`` can be t
ought as giving the slice which must be
attributes. ``indptr`` can be t
hought of as giving the slice which
applied to the other attribute in order to get each row of the
must be
applied to the other attribute in order to get each row of the
matrix. In other words, ``slice(indptr[i], indptr[i+1])`` correspond
matrix. In other words, ``slice(indptr[i], indptr[i+1])`` correspond
s
to the slice needed to find the i-th row of the matrix in the ``data``
to the slice needed to find the i-th row of the matrix in the ``data``
and
in the
``indices`` fields.
and ``indices`` fields.
The following example builds a matrix and returns its rows. It prints
The following example builds a matrix and returns its rows. It prints
the i-th row, i.e. a list of indices in the row and their
corresponding value
the i-th row, i.e. a list of indices in the row and their
in the second list.
corresponding value
in the second list.
>>> data = np.asarray([7, 8, 9])
>>> data = np.asarray([7, 8, 9])
>>> indices = np.asarray([0, 1, 2])
>>> indices = np.asarray([0, 1, 2])
...
@@ -120,7 +120,7 @@ List of Implemented Operations
...
@@ -120,7 +120,7 @@ List of Implemented Operations
- Moving from and to sparse
- Moving from and to sparse
- :class:`DenseFromSparse <theano.sparse.basic.DenseFromSparse>` and ``dense_from_sparse``.
- :class:`DenseFromSparse <theano.sparse.basic.DenseFromSparse>` and ``dense_from_sparse``.
Both grad are implemented. Structured by default.
Both grad
s
are implemented. Structured by default.
- :class:`SparseFromDense <theano.sparse.basic.SparseFromDense>` and ``csr_from_dense``, ``csc_from_dense``.
- :class:`SparseFromDense <theano.sparse.basic.SparseFromDense>` and ``csr_from_dense``, ``csc_from_dense``.
The grad implemented is structured.
The grad implemented is structured.
- Theano SparseVariable object have a method ``toarray()`` that is the same as ``dense_from_sparse``.
- Theano SparseVariable object have a method ``toarray()`` that is the same as ``dense_from_sparse``.
...
@@ -201,51 +201,55 @@ List of Implemented Operations
...
@@ -201,51 +201,55 @@ List of Implemented Operations
- One of the inputs must be sparse, the other sparse or dense.
- One of the inputs must be sparse, the other sparse or dense.
- The grad implemented is regular.
- The grad implemented is regular.
- No C code for perform and no C code for grad.
- No C code for perform and no C code for grad.
- Return a dense for perform and a dense for grad.
- Return
s
a dense for perform and a dense for grad.
- :class:`StructuredDot <theano.sparse.basic.StructuredDot>`
- :class:`StructuredDot <theano.sparse.basic.StructuredDot>`
and :func:`structured_dot <theano.sparse.basic.structured_dot>`.
and :func:`structured_dot <theano.sparse.basic.structured_dot>`.
- The first input is sparse, the second can be sparse or dense.
- The first input is sparse, the second can be sparse or dense.
- The grad implemented is structured.
- The grad implemented is structured.
- C code for perform and grad.
- C code for perform and grad.
- Return a dense for perforn and a sparse for grad.
- It returns a sparse output if both inputs are sparse and
dense one if one of the inputs is dense.
- Returns a sparse grad for sparse inputs and dense grad for
dense inputs.
- :class:`TrueDot <theano.sparse.basic.TrueDot>` and
- :class:`TrueDot <theano.sparse.basic.TrueDot>` and
:func:`true_dot <theano.sparse.basic.true_dot>`.
:func:`true_dot <theano.sparse.basic.true_dot>`.
- The first input is sparse, the second can be sparse or dense.
- The first input is sparse, the second can be sparse or dense.
- The grad implemented is regular.
- The grad implemented is regular.
- No C code for perform and no C code for grad.
- No C code for perform and no C code for grad.
- Return a Sparse for perform and a Sparse for grad.
- Returns a Sparse.
- Flags trough constructor can change the output of
- The gradient returns a Sparse for sparse inputs and by
grad to be dense if the second input of the op is dense.
default a dense for dense inputs. The parameter
``grad_preserves_dense`` can be set to False to return a
sparse grad for dense inputs.
- :class:`SamplingDot <theano.sparse.basic.SamplingDot>` and
- :class:`SamplingDot <theano.sparse.basic.SamplingDot>` and
``sampling_dot``.
``sampling_dot``.
- Both input must be dense.
- Both input
s
must be dense.
- The grad implemented is structured for `p`.
- The grad implemented is structured for `p`.
- Sample of the dot and sample of the gradient.
- Sample of the dot and sample of the gradient.
- C code for perform but not for grad.
- C code for perform but not for grad.
- Return sparse for perform and grad.
- Return
s
sparse for perform and grad.
- :class:`Usmm <theano.sparse.basic.Usmm>` and ``usmm``.
- :class:`Usmm <theano.sparse.basic.Usmm>` and ``usmm``.
- You *shouldn't* insert this op yourself!
- You *shouldn't* insert this op yourself!
- There is optimization that transform a
- There is
an
optimization that transform a
:class:`Dot <theano.sparse.basic.Dot>` to ``Usmm`` when possible.
:class:`Dot <theano.sparse.basic.Dot>` to ``Usmm`` when possible.
- This op is the equivalent of gemm for sparse dot.
- This op is the equivalent of gemm for sparse dot.
- There is no grad implemented for this op and this is not needed as
- There is no grad implemented for this op.
you don't insert it yourself.
- One of the inputs must be sparse, the other sparse or dense.
- One of the inputs must be sparse, the other sparse or dense.
- Return
a dense for perform
- Return
s a dense from perform.
- Slice Operations
- Slice Operations
- sparse_variable[N, N], return a tensor scalar.
- sparse_variable[N, N], return
s
a tensor scalar.
There is no grad implemented for this operation.
There is no grad implemented for this operation.
- sparse_variable[M:N, O:P], return a sparse matrix
- sparse_variable[M:N, O:P], return
s
a sparse matrix
There is no grad implemented for this operation.
There is no grad implemented for this operation.
- Sparse variable
don't support [M, N:O] and [M:N, O] as we don't support sparse vector
- Sparse variable
s don't support [M, N:O] and [M:N, O] as we don't
and returning a sparse matrix would break the numpy interface.
support sparse vectors and returning a sparse matrix would break
Use [M:M+1, N:O] and [M:N, O:O+1] instead.
the numpy interface.
Use [M:M+1, N:O] and [M:N, O:O+1] instead.
- :class:`Diag <theano.sparse.basic.Diag>` and ``diag``.
- :class:`Diag <theano.sparse.basic.Diag>` and ``diag``.
The grad implemented is regular.
The grad implemented is regular.
...
...
doc/library/tensor/raw_random.txt
浏览文件 @
d3ca3329
差异被折叠。
点击展开。
doc/tutorial/examples.txt
浏览文件 @
d3ca3329
...
@@ -5,13 +5,13 @@
...
@@ -5,13 +5,13 @@
More Examples
More Examples
=============
=============
At this point it would be wise to begin familiarizing yourself
At this point it would be wise to begin familiarizing yourself
more
more systematically with Theano's fundamental objects and operations by browsing
systematically with Theano's fundamental objects and operations by
this section of the library: :ref:`libdoc_basic_tensor`.
browsing
this section of the library: :ref:`libdoc_basic_tensor`.
As the tutorial unfolds, you should also gradually acquaint yourself
with the other
As the tutorial unfolds, you should also gradually acquaint yourself
relevant areas of the library and with the relevant subjects of the documentation
with the other relevant areas of the library and with the relevant
entrance page.
subjects of the documentation
entrance page.
Logistic Function
Logistic Function
...
@@ -30,13 +30,13 @@ the logistic curve, which is given by:
...
@@ -30,13 +30,13 @@ the logistic curve, which is given by:
A plot of the logistic function, with x on the x-axis and s(x) on the
A plot of the logistic function, with x on the x-axis and s(x) on the
y-axis.
y-axis.
You want to compute the function :ref:`elementwise <libdoc_tensor_elementwise>` on matrices of
You want to compute the function :ref:`elementwise
doubles, which means that you want to apply this function to each
<libdoc_tensor_elementwise>` on matrices of doubles, which means that
individual element of the matrix.
you want to apply this function to each individual element of the
matrix.
Well, what you do is this:
Well, what you do is this:
.. If you modify this code, also change :
.. If you modify this code, also change :
.. theano/tests/test_tutorial.py:T_examples.test_examples_1
.. theano/tests/test_tutorial.py:T_examples.test_examples_1
...
@@ -450,6 +450,10 @@ Other Random Distributions
...
@@ -450,6 +450,10 @@ Other Random Distributions
There are :ref:`other distributions implemented <libdoc_tensor_raw_random>`.
There are :ref:`other distributions implemented <libdoc_tensor_raw_random>`.
Other Implementations
---------------------
There is 2 other implementations based on :class:`CURAND <theano.sandbox.cuda.rng_curand>` and :ref:`MRG31k3p <libdoc_rng_mrg>`
.. _logistic_regression:
.. _logistic_regression:
...
@@ -457,7 +461,8 @@ There are :ref:`other distributions implemented <libdoc_tensor_raw_random>`.
...
@@ -457,7 +461,8 @@ There are :ref:`other distributions implemented <libdoc_tensor_raw_random>`.
A Real Example: Logistic Regression
A Real Example: Logistic Regression
===================================
===================================
The preceding elements are featured in this more realistic example. It will be used repeatedly.
The preceding elements are featured in this more realistic example.
It will be used repeatedly.
.. code-block:: python
.. code-block:: python
...
...
doc/tutorial/multi_cores.txt
浏览文件 @
d3ca3329
...
@@ -2,30 +2,43 @@
...
@@ -2,30 +2,43 @@
Multi cores support in Theano
Multi cores support in Theano
=============================
=============================
Parallel element wise op with openmp
BLAS operation
==============
======================
==============
Beacuse element wise ops work on every tensor entry indipedently they can be
BLAS is an interface for some mathematic operations between two
easly parallelized using openmp.
vectors, a vector and a matrix or two matrices (e.g. the dot product
between vector/matrix and matrix/matrix). Many different
implementations of that interface exist and some of them are
parallelized.
To use openmp you must set the openmp flag in Theano configuration.
Theano tries to use that interface as frequently as possible for
performance reasons. So if Theano links to a parallel implementation,
those operations will run in parallel in Theano.
Yuo can use the flag openmp_elemwise_minsize to set the minimum tensor siz
e
The most frequent way to control the number of threads used is via th
e
for which the operation is parallelized because for short tensor using opemp
``OMP_NUM_THREADS`` environment variable. Set it to the number of threads
can slow down the operation
.
you want to use before starting the python process
.
If it is no specified the default value (200000) is used.
For simple(fast) operation you can obtain a speed up for very long tensor
Parallel element wise ops with OpenMP
while for more complex operation you ca obtain a good speed up also for not
=====================================
too long tensor.
There is a script (elemwise_openmp_speedup.py in theano/misc/) which you can
use to choose that value for your machine.
The script run two elemwise operation (a fast and a slow one) for a vector of
size openmp_elemwise_minsize with and without openmp and show the time
difference between the two cases.
Because element wise ops work on every tensor entry independently they
can be easily parallelized using OpenMP.
To use OpenMP you must set the OpenMP flag in Theano configuration.
You can use the flag ``openmp_elemwise_minsize`` to set the minimum
tensor size for which the operation is parallelized because for short
tensors using OpenMP can slow down the operation. The default value is
``200000``.
For simple(fast) operation you can obtain a speed up with very large
tensors while for more complex operation you can obtain a good speed
up also for smaller tensor.
There is a script ``elemwise_openmp_speedup.py`` in ``theano/misc/``
which you can use to tune the value of ``openmp_elemwise_minsize`` for
your machine. The script runs two elemwise operations (a fast one and
a slow one) for a vector of size ``openmp_elemwise_minsize`` with and
without OpenMP and shows the time difference between the cases.
theano/misc/check_blas.py
浏览文件 @
d3ca3329
...
@@ -205,6 +205,7 @@ if __name__ == "__main__":
...
@@ -205,6 +205,7 @@ if __name__ == "__main__":
gpu
gpu
K20m/ECC 0.07s
K20m/ECC 0.07s
K20/NOECC 0.07s
K20/NOECC 0.07s
M2090 0.19s
C2075 0.25s
C2075 0.25s
M2075 0.25s
M2075 0.25s
M2070 0.25s 0.27s 0.32s
M2070 0.25s 0.27s 0.32s
...
...
theano/sandbox/cuda/tests/test_type.py
浏览文件 @
d3ca3329
...
@@ -12,10 +12,13 @@ if cuda_available:
...
@@ -12,10 +12,13 @@ if cuda_available:
# >>> with open('CudaNdarray.pkl', 'wb') as fp:
# >>> with open('CudaNdarray.pkl', 'wb') as fp:
# >>> cPickle.dump(theano.sandbox.cuda.CudaNdarray(np.array([-42.0], dtype=np.float32)), fp)
# >>> cPickle.dump(theano.sandbox.cuda.CudaNdarray(np.array([-42.0], dtype=np.float32)), fp)
def
test_unpickle_flag_is_false_by_default
():
def
test_unpickle_flag_is_false_by_default
():
assert
not
config
.
experimental
.
unpickle_gpu_on_cpu
,
"Config flag experimental.unpickle_gpu_on_cpu is "
\
assert
not
config
.
experimental
.
unpickle_gpu_on_cpu
,
(
+
"set to true. Make sure the default value stays false "
\
"Config flag experimental.unpickle_gpu_on_cpu is "
+
"and that you have not set the flag manually."
"set to true. Make sure the default value stays false "
"and that you have not set the flag manually."
)
def
test_unpickle_cudandarray_as_numpy_ndarray_flag0
():
def
test_unpickle_cudandarray_as_numpy_ndarray_flag0
():
oldflag
=
config
.
experimental
.
unpickle_gpu_on_cpu
oldflag
=
config
.
experimental
.
unpickle_gpu_on_cpu
...
...
theano/sandbox/rng_mrg.py
浏览文件 @
d3ca3329
...
@@ -734,9 +734,11 @@ class MRG_RandomStreams(object):
...
@@ -734,9 +734,11 @@ class MRG_RandomStreams(object):
:param low: Lower bound of the interval on which values are sampled.
:param low: Lower bound of the interval on which values are sampled.
If the ``dtype`` arg is provided, ``low`` will be cast into dtype.
If the ``dtype`` arg is provided, ``low`` will be cast into dtype.
This bound is excluded.
:param high: Higher bound of the interval on which values are sampled.
:param high: Higher bound of the interval on which values are sampled.
If the ``dtype`` arg is provided, ``high`` will be cast into dtype.
If the ``dtype`` arg is provided, ``high`` will be cast into dtype.
This bound is excluded.
:param size: Can be a list of integer or Theano variable
:param size: Can be a list of integer or Theano variable
(ex: the shape of other Theano Variable)
(ex: the shape of other Theano Variable)
...
...
theano/scalar/basic.py
浏览文件 @
d3ca3329
...
@@ -869,7 +869,8 @@ class ScalarOp(Op):
...
@@ -869,7 +869,8 @@ class ScalarOp(Op):
return
self
.
name
return
self
.
name
else
:
else
:
param
=
[(
k
,
v
)
for
k
,
v
in
self
.
__dict__
.
items
()
param
=
[(
k
,
v
)
for
k
,
v
in
self
.
__dict__
.
items
()
if
k
not
in
[
"name"
,
"_op_use_c_code"
]]
if
k
not
in
[
"name"
,
"_op_use_c_code"
,
"output_types_preference"
]]
if
param
:
if
param
:
return
"
%
s{
%
s}"
%
(
self
.
__class__
.
__name__
,
return
"
%
s{
%
s}"
%
(
self
.
__class__
.
__name__
,
", "
.
join
(
"
%
s=
%
s"
%
(
k
,
v
)
", "
.
join
(
"
%
s=
%
s"
%
(
k
,
v
)
...
...
theano/sparse/basic.py
浏览文件 @
d3ca3329
...
@@ -2623,11 +2623,14 @@ class TrueDot(gof.op.Op):
...
@@ -2623,11 +2623,14 @@ class TrueDot(gof.op.Op):
self
.
grad_preserves_dense
=
grad_preserves_dense
self
.
grad_preserves_dense
=
grad_preserves_dense
def
__eq__
(
self
,
other
):
def
__eq__
(
self
,
other
):
return
(
type
(
self
)
==
type
(
other
)
and
# The grad_preserves_dense attribute doesn't change the
self
.
grad_preserves_dense
==
other
.
grad_preserves_dense
)
# execution behavior. To let the optimizer merge nodes with
# different values of this attribute we shouldn't compare it
# here.
return
type
(
self
)
==
type
(
other
)
def
__hash__
(
self
):
def
__hash__
(
self
):
return
hash
(
type
(
self
))
^
hash
(
self
.
grad_preserves_dense
)
return
hash
(
type
(
self
))
def
__ne__
(
self
,
other
):
def
__ne__
(
self
,
other
):
return
not
(
self
==
other
)
return
not
(
self
==
other
)
...
@@ -2712,15 +2715,17 @@ class TrueDot(gof.op.Op):
...
@@ -2712,15 +2715,17 @@ class TrueDot(gof.op.Op):
def
true_dot
(
x
,
y
,
grad_preserves_dense
=
True
):
def
true_dot
(
x
,
y
,
grad_preserves_dense
=
True
):
"""
"""
Operation for efficiently calculating the dot product when
Operation for efficiently calculating the dot product when
one or all operands
is sparse. Supported format
are CSC and CSR.
one or all operands
are sparse. Supported formats
are CSC and CSR.
The output of the operation is sparse.
The output of the operation is sparse.
:param x:
Matrix
variable.
:param x:
Sparse matrix or 2d tensor
variable.
:param y:
Matrix
variable.
:param y:
Sparse matrix or 2d tensor
variable.
:param grad_preserves_dense: if True
and one on the input is dense,
:param grad_preserves_dense: if True
(default), makes the grad of
make the output den
se.
dense inputs dense. Otherwise the grad is always spar
se.
:return: The dot product `x`.`y` in a sparse format.
:return: The dot product `x`.`y` in a sparse format.
:note: one of ``x`` or ``y`` must be sparse.
"""
"""
# TODO
# TODO
# Maybe the triple-transposition formulation
# Maybe the triple-transposition formulation
...
...
theano/tensor/nnet/Conv3D.py
浏览文件 @
d3ca3329
...
@@ -562,9 +562,13 @@ conv3D = Conv3D()
...
@@ -562,9 +562,13 @@ conv3D = Conv3D()
:note: The order of dimensions does not correspond to the one in `conv2d`.
:note: The order of dimensions does not correspond to the one in `conv2d`.
This is for optimization.
This is for optimization.
:note: The GPU implementation is very slow. You are better to use
:note: The GPU implementation is very slow. You should use
:func:`conv3d2d <theano.tensor.nnet.conv3d2d.conv3d>` that is faster
:func:`conv3d2d <theano.tensor.nnet.conv3d2d.conv3d>` for a GPU
on GPU.
graph instead.
:see: Someone made a script that shows how to swap the axes between
both 3d convolution implementations in Theano. See the last
`attachment <https://groups.google.com/d/msg/theano-users/1S9_bZgHxVw/0cQR9a4riFUJ>`_.
"""
"""
...
...
theano/tensor/nnet/conv3d2d.py
浏览文件 @
d3ca3329
...
@@ -178,6 +178,10 @@ def conv3d(signals, filters,
...
@@ -178,6 +178,10 @@ def conv3d(signals, filters,
Another way to define signals: (batch, time, in channel, row, column)
Another way to define signals: (batch, time, in channel, row, column)
Another way to define filters: (out channel,time,in channel, row, column)
Another way to define filters: (out channel,time,in channel, row, column)
:see: Someone made a script that shows how to swap the axes between
both 3d convolution implementations in Theano. See the last
`attachment <https://groups.google.com/d/msg/theano-users/1S9_bZgHxVw/0cQR9a4riFUJ>`_.
"""
"""
if
isinstance
(
border_mode
,
str
):
if
isinstance
(
border_mode
,
str
):
...
...
theano/tensor/raw_random.py
浏览文件 @
d3ca3329
...
@@ -576,11 +576,11 @@ def random_integers(random_state, size=None, low=0, high=1, ndim=None,
...
@@ -576,11 +576,11 @@ def random_integers(random_state, size=None, low=0, high=1, ndim=None,
def
choice_helper
(
random_state
,
a
,
replace
,
p
,
size
):
def
choice_helper
(
random_state
,
a
,
replace
,
p
,
size
):
"""
"""Helper function to draw random numbers using numpy's choice function.
Helper function to draw random numbers using numpy's choice function.
This is a generalization of numpy.random.choice to the case where `a`,
This is a generalization of numpy.random.choice that coerces
`replace` and `p` are tensors.
`replace` to a bool and replaces `p` with None when p is a vector
of 0 elements.
"""
"""
if
a
.
ndim
>
1
:
if
a
.
ndim
>
1
:
raise
ValueError
(
'a.ndim (
%
i) must be 0 or 1'
%
a
.
ndim
)
raise
ValueError
(
'a.ndim (
%
i) must be 0 or 1'
%
a
.
ndim
)
...
@@ -622,16 +622,6 @@ def choice(random_state, size=None, a=2, replace=True, p=None, ndim=None,
...
@@ -622,16 +622,6 @@ def choice(random_state, size=None, a=2, replace=True, p=None, ndim=None,
broadcastable
=
bcast
))
broadcastable
=
bcast
))
return
op
(
random_state
,
size
,
a
,
replace
,
p
)
return
op
(
random_state
,
size
,
a
,
replace
,
p
)
def
poisson_helper
(
random_state
,
lam
,
size
):
"""
Helper function to draw random numbers using numpy's poisson function.
This is a generalization of numpy.random.poisson to the case where
`lam` is a tensor.
"""
return
random_state
.
poisson
(
lam
,
size
)
def
poisson
(
random_state
,
size
=
None
,
lam
=
1.0
,
ndim
=
None
,
dtype
=
'int64'
):
def
poisson
(
random_state
,
size
=
None
,
lam
=
1.0
,
ndim
=
None
,
dtype
=
'int64'
):
"""
"""
Draw samples from a Poisson distribution.
Draw samples from a Poisson distribution.
...
@@ -652,7 +642,7 @@ def poisson(random_state, size=None, lam=1.0, ndim=None, dtype='int64'):
...
@@ -652,7 +642,7 @@ def poisson(random_state, size=None, lam=1.0, ndim=None, dtype='int64'):
ndim
,
size
,
bcast
=
_infer_ndim_bcast
(
ndim
,
size
)
ndim
,
size
,
bcast
=
_infer_ndim_bcast
(
ndim
,
size
)
op
=
RandomFunction
(
poisson_helper
,
tensor
.
TensorType
(
dtype
=
dtype
,
op
=
RandomFunction
(
"poisson"
,
tensor
.
TensorType
(
dtype
=
dtype
,
broadcastable
=
bcast
))
broadcastable
=
bcast
))
return
op
(
random_state
,
size
,
lam
)
return
op
(
random_state
,
size
,
lam
)
...
@@ -668,6 +658,9 @@ def permutation_helper(random_state, n, shape):
...
@@ -668,6 +658,9 @@ def permutation_helper(random_state, n, shape):
If you wish to perform a permutation of the elements of an existing vector,
If you wish to perform a permutation of the elements of an existing vector,
see shuffle_row_elements.
see shuffle_row_elements.
This is a generalization of numpy.random.permutation to tensors.
Otherwise it behaves the same.
"""
"""
# n should be a 0-dimension array
# n should be a 0-dimension array
assert
n
.
shape
==
()
assert
n
.
shape
==
()
...
@@ -680,7 +673,7 @@ def permutation_helper(random_state, n, shape):
...
@@ -680,7 +673,7 @@ def permutation_helper(random_state, n, shape):
shape
=
()
shape
=
()
out_shape
=
list
(
shape
)
out_shape
=
list
(
shape
)
out_shape
.
append
(
n
)
out_shape
.
append
(
n
)
out
=
numpy
.
zeros
(
out_shape
,
int
)
out
=
numpy
.
empty
(
out_shape
,
int
)
for
i
in
numpy
.
ndindex
(
*
shape
):
for
i
in
numpy
.
ndindex
(
*
shape
):
out
[
i
]
=
random_state
.
permutation
(
n
)
out
[
i
]
=
random_state
.
permutation
(
n
)
...
@@ -869,7 +862,7 @@ class RandomStreamsBase(object):
...
@@ -869,7 +862,7 @@ class RandomStreamsBase(object):
def
binomial
(
self
,
size
=
None
,
n
=
1
,
p
=
0.5
,
ndim
=
None
,
dtype
=
'int64'
,
def
binomial
(
self
,
size
=
None
,
n
=
1
,
p
=
0.5
,
ndim
=
None
,
dtype
=
'int64'
,
prob
=
None
):
prob
=
None
):
"""
"""
Sample n times with probability of success p
rob for each trial,
Sample n times with probability of success p
for each trial and
return the number of successes.
return the number of successes.
If the size argument is ambiguous on the number of dimensions,
If the size argument is ambiguous on the number of dimensions,
...
...
编写
预览
Markdown
格式
0%
重试
或
添加新文件
添加附件
取消
您添加了
0
人
到此讨论。请谨慎行事。
请先完成此评论的编辑!
取消
请
注册
或者
登录
后发表评论