def get_output_size(self, in_dims): if len(in_dims) == 1: return [in_dims[0]] return [Dim.unknown()]
def get_output_size(self, in_dims): if self.indicated_outputs: return self.indicated_outputs if len(in_dims) == 1: return [in_dims[0]] return [Dim.unknown()]