提交 16fce528 authored 作者: Brandon T. Willard's avatar Brandon T. Willard 提交者: Brandon T. Willard

Fix Scan.truncate_gradient default value and type

上级 d38552d9
......@@ -587,7 +587,7 @@ class Scan(Op, ScanMethodsMixin, HasInnerGraph):
info: ScanInfo,
mode: Optional[Mode] = None,
typeConstructor: Optional[TensorConstructorType] = None,
truncate_gradient: bool = False,
truncate_gradient: int = -1,
name: Optional[str] = None,
as_while: bool = False,
profile: Optional[Union[str, bool]] = None,
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论