提交 e4ca40f9 authored 作者: Brandon T. Willard's avatar Brandon T. Willard 提交者: Ricardo Vieira

Use Type.filter in NumbaLinker.output_filter

上级 ba51e7d2
......@@ -17,7 +17,7 @@ class NumbaLinker(JITLinker):
if not isinstance(var, np.ndarray) and isinstance(
var.type, pytensor.tensor.TensorType
):
return np.asarray(out, dtype=var.type.dtype)
return var.type.filter(out, allow_downcast=True)
return out
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论