提交 a486ae68 authored 作者: Pascal Lamblin's avatar Pascal Lamblin

In Scan.infer_shape, the shapes of node.inputs are valid values to use.

上级 b3965fab
...@@ -628,9 +628,10 @@ class Scan(Op): ...@@ -628,9 +628,10 @@ class Scan(Op):
inputs = self.inputs, inputs = self.inputs,
input_shapes = inner_ins_shapes) input_shapes = inner_ins_shapes)
# Will be used to check if outs_shape can be expressed without using # Will be used to check if outs_shape can be expressed without using
# variables in self.inputs # variables in self.inputs.
# The shapes of node.inputs are valid.
validator = scan_utils.Validator( validator = scan_utils.Validator(
valid = [], valid = input_shapes,
invalid = self.inputs, invalid = self.inputs,
valid_equivalent = out_equivalent) valid_equivalent = out_equivalent)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论