提交 a0c79ca5 authored 作者: Frederic's avatar Frederic

Add theano.tensor.blas.{gemv,gemm} as a shortcut.

上级 fe38ef37
...@@ -246,7 +246,8 @@ class Gemv(Op): ...@@ -246,7 +246,8 @@ class Gemv(Op):
gemv_no_inplace = Gemv(inplace=False) gemv_no_inplace = Gemv(inplace=False)
gemv_inplace = Gemv(inplace=True) gemv_inplace = Gemv(inplace=True)
# For the user interface. Opt will make them inplace later
gemv = gemv_no_inplace
class Ger(Op): class Ger(Op):
""" """
...@@ -991,6 +992,8 @@ class Gemm(GemmRelated): ...@@ -991,6 +992,8 @@ class Gemm(GemmRelated):
gemm_inplace = Gemm(inplace=True) gemm_inplace = Gemm(inplace=True)
gemm_no_inplace = Gemm(inplace=False) gemm_no_inplace = Gemm(inplace=False)
# For the user interface. Theano optimization will make them inplace
gemm = gemm_no_inplace
pprint.assign(gemm_inplace, FunctionPrinter('gemm_inplace')) pprint.assign(gemm_inplace, FunctionPrinter('gemm_inplace'))
pprint.assign(gemm_no_inplace, FunctionPrinter('gemm_no_inplace')) pprint.assign(gemm_no_inplace, FunctionPrinter('gemm_no_inplace'))
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论