Skip to content

P
pytensor
提交 3aa69e40 authored 作者: desjagui@atchoum.iro.umontreal.ca's avatar desjagui@atchoum.iro.umontreal.ca
浏览文件
操作

Output of TensorDot has "broadcastable" dimensions set to [None]*<dim> instead

of [False]*<dim>
上级 6bca7720
隐藏空白字符变更
内嵌 并排
正在显示 包含 1 行增加 和 1 行删除
theano/tensor/basic.py
浏览文件 @ 3aa69e40
...@@ -2043,7 +2043,7 @@ class TensorDot(Op): ...@@ -2043,7 +2043,7 @@ class TensorDot(Op):
axesdim, x.type.ndim, y.type.ndim) axesdim, x.type.ndim, y.type.ndim)
outdim = x.type.ndim + y.type.ndim - 2*axesdim outdim = x.type.ndim + y.type.ndim - 2*axesdim
output = tensor(dtype=x.dtype, broadcastable=[None]*outdim); output = tensor(dtype=x.dtype, broadcastable=[False]*outdim);
return Apply(self, inputs=[x,y], outputs=[output,]) return Apply(self, inputs=[x,y], outputs=[output,])
def perform(self, node, (x, y), (z,)): def perform(self, node, (x, y), (z,)):
......
Markdown 格式
0% 或
您添加了 0 人 到此讨论。请谨慎行事。
请先完成此评论的编辑！
请 注册 或者 后发表评论