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