提交 7b3fd950 authored 作者: Frederic's avatar Frederic

add info in docstring.

上级 6c60f4b8
...@@ -3314,7 +3314,7 @@ class StructuredDot(gof.Op): ...@@ -3314,7 +3314,7 @@ class StructuredDot(gof.Op):
:param a: A sparse matrix. :param a: A sparse matrix.
:param b: A sparse or dense matrix. :param b: A sparse or dense matrix.
:return: The dot product of `a` and `b`. :return: The dot product of `a` and `b` as a dense matrix.
:note: :note:
- The grad implemented is structured. - The grad implemented is structured.
...@@ -4406,8 +4406,8 @@ class Dot(gof.op.Op): ...@@ -4406,8 +4406,8 @@ class Dot(gof.op.Op):
one or all operands is sparse. Supported format are CSC and CSR. one or all operands is sparse. Supported format are CSC and CSR.
The output of the operation is dense. The output of the operation is dense.
:param x: Matrix variable. :param x: sparse or dense matrix variable.
:param y: Matrix variable. :param y: sparse or dense matrix variable.
:return: The dot product `x`.`y` in a dense format. :return: The dot product `x`.`y` in a dense format.
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论