提交 603c7167 authored 作者: Brandon T. Willard's avatar Brandon T. Willard

Maintain dtype in triu

上级 2f7eb1d2
......@@ -2979,7 +2979,9 @@ def triu(m, k=0):
tril : Lower triangle of an array.
"""
return m * (1 - tri(m.shape[0], m.shape[1], k=k - 1, dtype=m.dtype))
return m * (
constant(1, dtype=m.dtype) - tri(m.shape[0], m.shape[1], k=k - 1, dtype=m.dtype)
)
class Eye(gof.Op):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论