提交 b3fa0e92 authored 作者: Pierre Luc Carrier's avatar Pierre Luc Carrier

Replace two transpose by a single one

上级 4ca66330
...@@ -756,9 +756,9 @@ class PushOutScanOutput(gof.Optimizer): ...@@ -756,9 +756,9 @@ class PushOutScanOutput(gof.Optimizer):
# Perform the Dot outside of scan # Perform the Dot outside of scan
if idx_matrix_input == 0: if idx_matrix_input == 0:
outer_dot_inputs = [outer_matrix_input, outer_dot_inputs = [outer_vector_input,
outer_vector_input.transpose()] outer_matrix_input.transpose()]
outer_dot_output = theano.tensor.dot(*outer_dot_inputs).transpose() outer_dot_output = theano.tensor.dot(*outer_dot_inputs)
else: # idx_matrix_input == 1 else: # idx_matrix_input == 1
outer_dot_inputs = [outer_vector_input, outer_dot_inputs = [outer_vector_input,
outer_matrix_input] outer_matrix_input]
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论