Unverified 提交 2d9a7aeb authored 作者: Arpan Jha's avatar Arpan Jha 提交者: GitHub

fix incorrect comment in StructuredDot.grad (#1890)

上级 3bd8ed9f
......@@ -1394,9 +1394,9 @@ class StructuredDot(Op):
out[0] = np.asarray(variable, str(variable.dtype))
def grad(self, inputs, gout):
# FIXME: It's not always true that b and g_out are dense.
# Python implementation (and numba) support sparse 'b' (and thus, 'g_out') as well.
# a is sparse, b is dense, g_out is dense
# NOTE: a is sparse; b and g_out are both dense or sparse.
# Python and Numba backends support sparse b and g_out.
# C backend only supports dense b and g_out.
# ga = g_out x b.T
# gb = a.T x g_out
(a, b) = inputs
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论