提交 4beb4dea authored 作者: Jeremiah Lowin's avatar Jeremiah Lowin

updated comment to match code

上级 c49c012c
...@@ -6983,13 +6983,14 @@ class Dot(Op): ...@@ -6983,13 +6983,14 @@ class Dot(Op):
# #
# for grad x: # for grad x:
# gradient is a tensordot over y and grad, summing out all but # gradient is a tensordot over y and grad, summing out all but
# the second-to-last dim of y and transposed such that the first # the second-to-last dim of y (if it exists) and transposed
# resulting dim goes last. # such that the first resulting dim goes last.
# #
# for grad y: # for grad y:
# gradient is a tensordot over x and grad, summing out all but # gradient is a tensordot over x and grad, summing out all but
# the last dim of x and transposed such that the first resulting # the last dim of x and transposed such that the first
# dim goes last. # resulting dim goes second-to-last (unless ydim == 1, when
# it goes last).
else: else:
x_axes0 = range(ydim-2) x_axes0 = range(ydim-2)
x_axes1 = range(xdim - 1, gdim) x_axes1 = range(xdim - 1, gdim)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论