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

Added kron to the doc.

上级 ca96d8f2
...@@ -17,4 +17,5 @@ API ...@@ -17,4 +17,5 @@ API
.. automodule:: theano.sandbox.linalg.ops .. automodule:: theano.sandbox.linalg.ops
:members: :members:
.. automodule:: theano.sandbox.linalg.kron
:members:
from kron import kron
from ops import (cholesky, matrix_inverse, solve, from ops import (cholesky, matrix_inverse, solve,
diag, extract_diag, alloc_diag, diag, extract_diag, alloc_diag,
det, psd, eig, eigh, det, psd, eig, eigh,
......
...@@ -2,6 +2,15 @@ from theano import tensor ...@@ -2,6 +2,15 @@ from theano import tensor
def kron(a, b): def kron(a, b):
""" Kronecker product
Same as scipy.linalg.kron(a, b) or numpy.kron(a, b)
:param a: array_like
:param b: array_like
:return: array_like with a.ndim + b.ndim - 2 dimensinos.
"""
a = tensor.as_tensor_variable(a) a = tensor.as_tensor_variable(a)
b = tensor.as_tensor_variable(b) b = tensor.as_tensor_variable(b)
if (a.ndim + b.ndim <= 2): if (a.ndim + b.ndim <= 2):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论