Skip to content

P
pytensor
提交 1944353c authored 作者: Ricardo Vieira's avatar Ricardo Vieira 提交者： Ricardo Vieira
浏览文件
操作

Simplify string expression for Constant Variables

上级 66934167
隐藏空白字符变更
内嵌 并排
正在显示 包含 97 行增加 和 105 行删除
pytensor/graph/basic.py
浏览文件 @ 1944353c
...@@ -763,13 +763,20 @@ class Constant(AtomicVariable[_TypeType]): ...@@ -763,13 +763,20 @@ class Constant(AtomicVariable[_TypeType]):
return (self.type, self.data) return (self.type, self.data)
def __str__(self): def __str__(self):
if self.name is not None: data_str = str(self.data)
return self.name if len(data_str) > 20:
else: data_str = data_str[:10].strip() + " ... " + data_str[-10:].strip()
name = str(self.data)
if len(name) > 20: if self.name is None:
name = name[:10] + "..." + name[-10:] return data_str
return f"{type(self).__name__}{{{name}}}"
return f"{self.name}{{{data_str}}}"
def __repr__(self):
data_str = repr(self.data)
if len(data_str) > 20:
data_str = data_str[:10].strip() + " ... " + data_str[-10:].strip()
return f"{type(self).__name__}({repr(self.type)}, data={data_str})"
def clone(self, **kwargs): def clone(self, **kwargs):
return self return self
......
pytensor/tensor/var.py
浏览文件 @ 1944353c
...@@ -1023,21 +1023,6 @@ class TensorConstant(TensorVariable, Constant[_TensorTypeType]): ...@@ -1023,21 +1023,6 @@ class TensorConstant(TensorVariable, Constant[_TensorTypeType]):
Constant.__init__(self, new_type, data, name) Constant.__init__(self, new_type, data, name)
def __str__(self):
unique_val = get_unique_value(self)
if unique_val is not None:
val = f"{self.data.shape} of {unique_val}"
else:
val = f"{self.data}"
if len(val) > 20:
val = val[:10].strip() + " ... " + val[-10:].strip()
if self.name is not None:
name = self.name
else:
name = "TensorConstant"
return f"{name}{{{val}}}"
def signature(self): def signature(self):
return TensorConstantSignature((self.type, self.data)) return TensorConstantSignature((self.type, self.data))
......
tests/graph/rewriting/test_basic.py
浏览文件 @ 1944353c
...@@ -166,7 +166,7 @@ class TestPatternNodeRewriter: ...@@ -166,7 +166,7 @@ class TestPatternNodeRewriter:
e = op1(op1(x, y), y) e = op1(op1(x, y), y)
g = FunctionGraph([y], [e]) g = FunctionGraph([y], [e])
OpKeyPatternNodeRewriter((op1, z, "1"), (op2, "1", z)).rewrite(g) OpKeyPatternNodeRewriter((op1, z, "1"), (op2, "1", z)).rewrite(g)
assert str(g) == "FunctionGraph(Op1(Op2(y, z), y))" assert str(g) == "FunctionGraph(Op1(Op2(y, z{2}), y))"
def test_constraints(self): def test_constraints(self):
x, y, z = MyVariable("x"), MyVariable("y"), MyVariable("z") x, y, z = MyVariable("x"), MyVariable("y"), MyVariable("z")
......
tests/scan/test_printing.py
浏览文件 @ 1944353c
...@@ -41,21 +41,21 @@ def test_debugprint_sitsot(): ...@@ -41,21 +41,21 @@ def test_debugprint_sitsot():
│ │ │ │ │ │ └─ Second [id L] │ │ │ │ │ │ └─ Second [id L]
│ │ │ │ │ │ ├─ A [id M] │ │ │ │ │ │ ├─ A [id M]
│ │ │ │ │ │ └─ ExpandDims{axis=0} [id N] │ │ │ │ │ │ └─ ExpandDims{axis=0} [id N]
│ │ │ │ │ │ └─ TensorConstant{1.0} [id O] │ │ │ │ │ │ └─ 1.0 [id O]
│ │ │ │ │ └─ ScalarConstant{0} [id P] │ │ │ │ │ └─ 0 [id P]
│ │ │ │ └─ Subtensor{i} [id Q] │ │ │ │ └─ Subtensor{i} [id Q]
│ │ │ │ ├─ Shape [id R] │ │ │ │ ├─ Shape [id R]
│ │ │ │ │ └─ Unbroadcast{0} [id J] │ │ │ │ │ └─ Unbroadcast{0} [id J]
│ │ │ │ │ └─ ··· │ │ │ │ │ └─ ···
│ │ │ │ └─ ScalarConstant{1} [id S] │ │ │ │ └─ 1 [id S]
│ │ │ ├─ Unbroadcast{0} [id J] │ │ │ ├─ Unbroadcast{0} [id J]
│ │ │ │ └─ ··· │ │ │ │ └─ ···
│ │ │ └─ ScalarFromTensor [id T] │ │ │ └─ ScalarFromTensor [id T]
│ │ │ └─ Subtensor{i} [id H] │ │ │ └─ Subtensor{i} [id H]
│ │ │ └─ ··· │ │ │ └─ ···
│ │ └─ A [id M] (outer_in_non_seqs-0) │ │ └─ A [id M] (outer_in_non_seqs-0)
│ └─ ScalarConstant{1} [id U] │ └─ 1 [id U]
└─ ScalarConstant{-1} [id V] └─ -1 [id V]
Inner graphs: Inner graphs:
...@@ -99,21 +99,21 @@ def test_debugprint_sitsot_no_extra_info(): ...@@ -99,21 +99,21 @@ def test_debugprint_sitsot_no_extra_info():
│ │ │ │ │ │ └─ Second [id L] │ │ │ │ │ │ └─ Second [id L]
│ │ │ │ │ │ ├─ A [id M] │ │ │ │ │ │ ├─ A [id M]
│ │ │ │ │ │ └─ ExpandDims{axis=0} [id N] │ │ │ │ │ │ └─ ExpandDims{axis=0} [id N]
│ │ │ │ │ │ └─ TensorConstant{1.0} [id O] │ │ │ │ │ │ └─ 1.0 [id O]
│ │ │ │ │ └─ ScalarConstant{0} [id P] │ │ │ │ │ └─ 0 [id P]
│ │ │ │ └─ Subtensor{i} [id Q] │ │ │ │ └─ Subtensor{i} [id Q]
│ │ │ │ ├─ Shape [id R] │ │ │ │ ├─ Shape [id R]
│ │ │ │ │ └─ Unbroadcast{0} [id J] │ │ │ │ │ └─ Unbroadcast{0} [id J]
│ │ │ │ │ └─ ··· │ │ │ │ │ └─ ···
│ │ │ │ └─ ScalarConstant{1} [id S] │ │ │ │ └─ 1 [id S]
│ │ │ ├─ Unbroadcast{0} [id J] │ │ │ ├─ Unbroadcast{0} [id J]
│ │ │ │ └─ ··· │ │ │ │ └─ ···
│ │ │ └─ ScalarFromTensor [id T] │ │ │ └─ ScalarFromTensor [id T]
│ │ │ └─ Subtensor{i} [id H] │ │ │ └─ Subtensor{i} [id H]
│ │ │ └─ ··· │ │ │ └─ ···
│ │ └─ A [id M] │ │ └─ A [id M]
│ └─ ScalarConstant{1} [id U] │ └─ 1 [id U]
└─ ScalarConstant{-1} [id V] └─ -1 [id V]
Inner graphs: Inner graphs:
...@@ -154,17 +154,17 @@ def test_debugprint_nitsot(): ...@@ -154,17 +154,17 @@ def test_debugprint_nitsot():
│ │ ├─ Shape [id E] │ │ ├─ Shape [id E]
│ │ │ └─ Subtensor{start:} [id F] 'coefficients[0:]' │ │ │ └─ Subtensor{start:} [id F] 'coefficients[0:]'
│ │ │ ├─ coefficients [id G] │ │ │ ├─ coefficients [id G]
│ │ │ └─ ScalarConstant{0} [id H] │ │ │ └─ 0 [id H]
│ │ └─ ScalarConstant{0} [id I] │ │ └─ 0 [id I]
│ └─ Subtensor{i} [id J] │ └─ Subtensor{i} [id J]
│ ├─ Shape [id K] │ ├─ Shape [id K]
│ │ └─ Subtensor{start:} [id L] │ │ └─ Subtensor{start:} [id L]
│ │ ├─ ARange{dtype='int64'} [id M] │ │ ├─ ARange{dtype='int64'} [id M]
│ │ │ ├─ TensorConstant{0} [id N] │ │ │ ├─ 0 [id N]
│ │ │ ├─ TensorConstant{10000} [id O] │ │ │ ├─ 10000 [id O]
│ │ │ └─ TensorConstant{1} [id P] │ │ │ └─ 1 [id P]
│ │ └─ ScalarConstant{0} [id Q] │ │ └─ 0 [id Q]
│ └─ ScalarConstant{0} [id R] │ └─ 0 [id R]
├─ Subtensor{:stop} [id S] (outer_in_seqs-0) ├─ Subtensor{:stop} [id S] (outer_in_seqs-0)
│ ├─ Subtensor{start:} [id F] 'coefficients[0:]' │ ├─ Subtensor{start:} [id F] 'coefficients[0:]'
│ │ └─ ··· │ │ └─ ···
...@@ -232,17 +232,17 @@ def test_debugprint_nested_scans(): ...@@ -232,17 +232,17 @@ def test_debugprint_nested_scans():
│ │ ├─ Shape [id E] │ │ ├─ Shape [id E]
│ │ │ └─ Subtensor{start:} [id F] 'c[0:]' │ │ │ └─ Subtensor{start:} [id F] 'c[0:]'
│ │ │ ├─ c [id G] │ │ │ ├─ c [id G]
│ │ │ └─ ScalarConstant{0} [id H] │ │ │ └─ 0 [id H]
│ │ └─ ScalarConstant{0} [id I] │ │ └─ 0 [id I]
│ └─ Subtensor{i} [id J] │ └─ Subtensor{i} [id J]
│ ├─ Shape [id K] │ ├─ Shape [id K]
│ │ └─ Subtensor{start:} [id L] │ │ └─ Subtensor{start:} [id L]
│ │ ├─ ARange{dtype='int64'} [id M] │ │ ├─ ARange{dtype='int64'} [id M]
│ │ │ ├─ TensorConstant{0} [id N] │ │ │ ├─ 0 [id N]
│ │ │ ├─ TensorConstant{10} [id O] │ │ │ ├─ 10 [id O]
│ │ │ └─ TensorConstant{1} [id P] │ │ │ └─ 1 [id P]
│ │ └─ ScalarConstant{0} [id Q] │ │ └─ 0 [id Q]
│ └─ ScalarConstant{0} [id R] │ └─ 0 [id R]
├─ Subtensor{:stop} [id S] (outer_in_seqs-0) ├─ Subtensor{:stop} [id S] (outer_in_seqs-0)
│ ├─ Subtensor{start:} [id F] 'c[0:]' │ ├─ Subtensor{start:} [id F] 'c[0:]'
│ │ └─ ··· │ │ └─ ···
...@@ -282,21 +282,21 @@ def test_debugprint_nested_scans(): ...@@ -282,21 +282,21 @@ def test_debugprint_nested_scans():
│ │ │ │ │ │ │ └─ Second [id BN] │ │ │ │ │ │ │ └─ Second [id BN]
│ │ │ │ │ │ │ ├─ *2-<Vector(float64, shape=(?,))> [id BO] -> [id W] (inner_in_non_seqs-0) │ │ │ │ │ │ │ ├─ *2-<Vector(float64, shape=(?,))> [id BO] -> [id W] (inner_in_non_seqs-0)
│ │ │ │ │ │ │ └─ ExpandDims{axis=0} [id BP] │ │ │ │ │ │ │ └─ ExpandDims{axis=0} [id BP]
│ │ │ │ │ │ │ └─ TensorConstant{1.0} [id BQ] │ │ │ │ │ │ │ └─ 1.0 [id BQ]
│ │ │ │ │ │ └─ ScalarConstant{0} [id BR] │ │ │ │ │ │ └─ 0 [id BR]
│ │ │ │ │ └─ Subtensor{i} [id BS] │ │ │ │ │ └─ Subtensor{i} [id BS]
│ │ │ │ │ ├─ Shape [id BT] │ │ │ │ │ ├─ Shape [id BT]
│ │ │ │ │ │ └─ Unbroadcast{0} [id BL] │ │ │ │ │ │ └─ Unbroadcast{0} [id BL]
│ │ │ │ │ │ └─ ··· │ │ │ │ │ │ └─ ···
│ │ │ │ │ └─ ScalarConstant{1} [id BU] │ │ │ │ │ └─ 1 [id BU]
│ │ │ │ ├─ Unbroadcast{0} [id BL] │ │ │ │ ├─ Unbroadcast{0} [id BL]
│ │ │ │ │ └─ ··· │ │ │ │ │ └─ ···
│ │ │ │ └─ ScalarFromTensor [id BV] │ │ │ │ └─ ScalarFromTensor [id BV]
│ │ │ │ └─ Subtensor{i} [id BJ] │ │ │ │ └─ Subtensor{i} [id BJ]
│ │ │ │ └─ ··· │ │ │ │ └─ ···
│ │ │ └─ *2-<Vector(float64, shape=(?,))> [id BO] -> [id W] (inner_in_non_seqs-0) (outer_in_non_seqs-0) │ │ │ └─ *2-<Vector(float64, shape=(?,))> [id BO] -> [id W] (inner_in_non_seqs-0) (outer_in_non_seqs-0)
│ │ └─ ScalarConstant{1} [id BW] │ │ └─ 1 [id BW]
│ └─ ScalarConstant{-1} [id BX] │ └─ -1 [id BX]
└─ ExpandDims{axis=0} [id BY] └─ ExpandDims{axis=0} [id BY]
└─ *1-<Scalar(int64, shape=())> [id BZ] -> [id U] (inner_in_seqs-1) └─ *1-<Scalar(int64, shape=())> [id BZ] -> [id U] (inner_in_seqs-1)
...@@ -325,17 +325,17 @@ def test_debugprint_nested_scans(): ...@@ -325,17 +325,17 @@ def test_debugprint_nested_scans():
│ │ ├─ Shape [id H] 5 │ │ ├─ Shape [id H] 5
│ │ │ └─ Subtensor{start:} [id I] 'c[0:]' 4 │ │ │ └─ Subtensor{start:} [id I] 'c[0:]' 4
│ │ │ ├─ c [id A] │ │ │ ├─ c [id A]
│ │ │ └─ ScalarConstant{0} [id J] │ │ │ └─ 0 [id J]
│ │ └─ ScalarConstant{0} [id K] │ │ └─ 0 [id K]
│ └─ Subtensor{i} [id L] 3 │ └─ Subtensor{i} [id L] 3
│ ├─ Shape [id M] 2 │ ├─ Shape [id M] 2
│ │ └─ Subtensor{start:} [id N] 1 │ │ └─ Subtensor{start:} [id N] 1
│ │ ├─ ARange{dtype='int64'} [id O] 0 │ │ ├─ ARange{dtype='int64'} [id O] 0
│ │ │ ├─ TensorConstant{0} [id P] │ │ │ ├─ 0 [id P]
│ │ │ ├─ TensorConstant{10} [id Q] │ │ │ ├─ 10 [id Q]
│ │ │ └─ TensorConstant{1} [id R] │ │ │ └─ 1 [id R]
│ │ └─ ScalarConstant{0} [id S] │ │ └─ 0 [id S]
│ └─ ScalarConstant{0} [id T] │ └─ 0 [id T]
├─ Subtensor{:stop} [id U] 11 (outer_in_seqs-0) ├─ Subtensor{:stop} [id U] 11 (outer_in_seqs-0)
│ ├─ Subtensor{start:} [id I] 'c[0:]' 4 │ ├─ Subtensor{start:} [id I] 'c[0:]' 4
│ │ └─ ··· │ │ └─ ···
...@@ -379,21 +379,21 @@ def test_debugprint_nested_scans(): ...@@ -379,21 +379,21 @@ def test_debugprint_nested_scans():
│ │ │ │ │ │ │ └─ Second [id BP] │ │ │ │ │ │ │ └─ Second [id BP]
│ │ │ │ │ │ │ ├─ *2-<Vector(float64, shape=(?,))> [id BA] (inner_in_non_seqs-0) │ │ │ │ │ │ │ ├─ *2-<Vector(float64, shape=(?,))> [id BA] (inner_in_non_seqs-0)
│ │ │ │ │ │ │ └─ ExpandDims{axis=0} [id BQ] │ │ │ │ │ │ │ └─ ExpandDims{axis=0} [id BQ]
│ │ │ │ │ │ │ └─ TensorConstant{1.0} [id BR] │ │ │ │ │ │ │ └─ 1.0 [id BR]
│ │ │ │ │ │ └─ ScalarConstant{0} [id BS] │ │ │ │ │ │ └─ 0 [id BS]
│ │ │ │ │ └─ Subtensor{i} [id BT] │ │ │ │ │ └─ Subtensor{i} [id BT]
│ │ │ │ │ ├─ Shape [id BU] │ │ │ │ │ ├─ Shape [id BU]
│ │ │ │ │ │ └─ Unbroadcast{0} [id BN] │ │ │ │ │ │ └─ Unbroadcast{0} [id BN]
│ │ │ │ │ │ └─ ··· │ │ │ │ │ │ └─ ···
│ │ │ │ │ └─ ScalarConstant{1} [id BV] │ │ │ │ │ └─ 1 [id BV]
│ │ │ │ ├─ Unbroadcast{0} [id BN] │ │ │ │ ├─ Unbroadcast{0} [id BN]
│ │ │ │ │ └─ ··· │ │ │ │ │ └─ ···
│ │ │ │ └─ ScalarFromTensor [id BW] │ │ │ │ └─ ScalarFromTensor [id BW]
│ │ │ │ └─ Subtensor{i} [id BL] │ │ │ │ └─ Subtensor{i} [id BL]
│ │ │ │ └─ ··· │ │ │ │ └─ ···
│ │ │ └─ *2-<Vector(float64, shape=(?,))> [id BA] (inner_in_non_seqs-0) (outer_in_non_seqs-0) │ │ │ └─ *2-<Vector(float64, shape=(?,))> [id BA] (inner_in_non_seqs-0) (outer_in_non_seqs-0)
│ │ └─ ScalarConstant{1} [id BX] │ │ └─ 1 [id BX]
│ └─ ScalarConstant{-1} [id BY] │ └─ -1 [id BY]
└─ ExpandDims{axis=0} [id BZ] └─ ExpandDims{axis=0} [id BZ]
└─ *1-<Scalar(int64, shape=())> [id Z] (inner_in_seqs-1) └─ *1-<Scalar(int64, shape=())> [id Z] (inner_in_seqs-1)
...@@ -432,17 +432,17 @@ def test_debugprint_mitsot(): ...@@ -432,17 +432,17 @@ def test_debugprint_mitsot():
expected_output = """Add [id A] expected_output = """Add [id A]
├─ Subtensor{start:} [id B] ├─ Subtensor{start:} [id B]
│ ├─ Scan{scan_fn, while_loop=False, inplace=none}.0 [id C] (outer_out_mit_sot-0) │ ├─ Scan{scan_fn, while_loop=False, inplace=none}.0 [id C] (outer_out_mit_sot-0)
│ │ ├─ TensorConstant{5} [id D] (n_steps) │ │ ├─ 5 [id D] (n_steps)
│ │ ├─ SetSubtensor{:stop} [id E] (outer_in_mit_sot-0) │ │ ├─ SetSubtensor{:stop} [id E] (outer_in_mit_sot-0)
│ │ │ ├─ AllocEmpty{dtype='int64'} [id F] │ │ │ ├─ AllocEmpty{dtype='int64'} [id F]
│ │ │ │ └─ Add [id G] │ │ │ │ └─ Add [id G]
│ │ │ │ ├─ TensorConstant{5} [id D] │ │ │ │ ├─ 5 [id D]
│ │ │ │ └─ Subtensor{i} [id H] │ │ │ │ └─ Subtensor{i} [id H]
│ │ │ │ ├─ Shape [id I] │ │ │ │ ├─ Shape [id I]
│ │ │ │ │ └─ Subtensor{:stop} [id J] │ │ │ │ │ └─ Subtensor{:stop} [id J]
│ │ │ │ │ ├─ <Vector(int64, shape=(?,))> [id K] │ │ │ │ │ ├─ <Vector(int64, shape=(?,))> [id K]
│ │ │ │ │ └─ ScalarConstant{2} [id L] │ │ │ │ │ └─ 2 [id L]
│ │ │ │ └─ ScalarConstant{0} [id M] │ │ │ │ └─ 0 [id M]
│ │ │ ├─ Subtensor{:stop} [id J] │ │ │ ├─ Subtensor{:stop} [id J]
│ │ │ │ └─ ··· │ │ │ │ └─ ···
│ │ │ └─ ScalarFromTensor [id N] │ │ │ └─ ScalarFromTensor [id N]
...@@ -451,23 +451,23 @@ def test_debugprint_mitsot(): ...@@ -451,23 +451,23 @@ def test_debugprint_mitsot():
│ │ └─ SetSubtensor{:stop} [id O] (outer_in_mit_sot-1) │ │ └─ SetSubtensor{:stop} [id O] (outer_in_mit_sot-1)
│ │ ├─ AllocEmpty{dtype='int64'} [id P] │ │ ├─ AllocEmpty{dtype='int64'} [id P]
│ │ │ └─ Add [id Q] │ │ │ └─ Add [id Q]
│ │ │ ├─ TensorConstant{5} [id D] │ │ │ ├─ 5 [id D]
│ │ │ └─ Subtensor{i} [id R] │ │ │ └─ Subtensor{i} [id R]
│ │ │ ├─ Shape [id S] │ │ │ ├─ Shape [id S]
│ │ │ │ └─ Subtensor{:stop} [id T] │ │ │ │ └─ Subtensor{:stop} [id T]
│ │ │ │ ├─ <Vector(int64, shape=(?,))> [id U] │ │ │ │ ├─ <Vector(int64, shape=(?,))> [id U]
│ │ │ │ └─ ScalarConstant{2} [id V] │ │ │ │ └─ 2 [id V]
│ │ │ └─ ScalarConstant{0} [id W] │ │ │ └─ 0 [id W]
│ │ ├─ Subtensor{:stop} [id T] │ │ ├─ Subtensor{:stop} [id T]
│ │ │ └─ ··· │ │ │ └─ ···
│ │ └─ ScalarFromTensor [id X] │ │ └─ ScalarFromTensor [id X]
│ │ └─ Subtensor{i} [id R] │ │ └─ Subtensor{i} [id R]
│ │ └─ ··· │ │ └─ ···
│ └─ ScalarConstant{2} [id Y] │ └─ 2 [id Y]
└─ Subtensor{start:} [id Z] └─ Subtensor{start:} [id Z]
├─ Scan{scan_fn, while_loop=False, inplace=none}.1 [id C] (outer_out_mit_sot-1) ├─ Scan{scan_fn, while_loop=False, inplace=none}.1 [id C] (outer_out_mit_sot-1)
│ └─ ··· │ └─ ···
└─ ScalarConstant{2} [id BA] └─ 2 [id BA]
Inner graphs: Inner graphs:
...@@ -519,28 +519,28 @@ def test_debugprint_mitmot(): ...@@ -519,28 +519,28 @@ def test_debugprint_mitmot():
│ │ │ │ │ │ │ │ └─ Second [id O] │ │ │ │ │ │ │ │ └─ Second [id O]
│ │ │ │ │ │ │ │ ├─ A [id P] │ │ │ │ │ │ │ │ ├─ A [id P]
│ │ │ │ │ │ │ │ └─ ExpandDims{axis=0} [id Q] │ │ │ │ │ │ │ │ └─ ExpandDims{axis=0} [id Q]
│ │ │ │ │ │ │ │ └─ TensorConstant{1.0} [id R] │ │ │ │ │ │ │ │ └─ 1.0 [id R]
│ │ │ │ │ │ │ └─ ScalarConstant{0} [id S] │ │ │ │ │ │ │ └─ 0 [id S]
│ │ │ │ │ │ └─ Subtensor{i} [id T] │ │ │ │ │ │ └─ Subtensor{i} [id T]
│ │ │ │ │ │ ├─ Shape [id U] │ │ │ │ │ │ ├─ Shape [id U]
│ │ │ │ │ │ │ └─ Unbroadcast{0} [id M] │ │ │ │ │ │ │ └─ Unbroadcast{0} [id M]
│ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ └─ ···
│ │ │ │ │ │ └─ ScalarConstant{1} [id V] │ │ │ │ │ │ └─ 1 [id V]
│ │ │ │ │ ├─ Unbroadcast{0} [id M] │ │ │ │ │ ├─ Unbroadcast{0} [id M]
│ │ │ │ │ │ └─ ··· │ │ │ │ │ │ └─ ···
│ │ │ │ │ └─ ScalarFromTensor [id W] │ │ │ │ │ └─ ScalarFromTensor [id W]
│ │ │ │ │ └─ Subtensor{i} [id K] │ │ │ │ │ └─ Subtensor{i} [id K]
│ │ │ │ │ └─ ··· │ │ │ │ │ └─ ···
│ │ │ │ └─ A [id P] (outer_in_non_seqs-0) │ │ │ │ └─ A [id P] (outer_in_non_seqs-0)
│ │ │ └─ ScalarConstant{0} [id X] │ │ │ └─ 0 [id X]
│ │ └─ TensorConstant{1} [id Y] │ │ └─ 1 [id Y]
│ ├─ Subtensor{:stop} [id Z] (outer_in_seqs-0) │ ├─ Subtensor{:stop} [id Z] (outer_in_seqs-0)
│ │ ├─ Subtensor{::step} [id BA] │ │ ├─ Subtensor{::step} [id BA]
│ │ │ ├─ Subtensor{:stop} [id BB] │ │ │ ├─ Subtensor{:stop} [id BB]
│ │ │ │ ├─ Scan{scan_fn, while_loop=False, inplace=none} [id F] (outer_out_sit_sot-0) │ │ │ │ ├─ Scan{scan_fn, while_loop=False, inplace=none} [id F] (outer_out_sit_sot-0)
│ │ │ │ │ └─ ··· │ │ │ │ │ └─ ···
│ │ │ │ └─ ScalarConstant{-1} [id BC] │ │ │ │ └─ -1 [id BC]
│ │ │ └─ ScalarConstant{-1} [id BD] │ │ │ └─ -1 [id BD]
│ │ └─ ScalarFromTensor [id BE] │ │ └─ ScalarFromTensor [id BE]
│ │ └─ Sub [id C] │ │ └─ Sub [id C]
│ │ └─ ··· │ │ └─ ···
...@@ -549,8 +549,8 @@ def test_debugprint_mitmot(): ...@@ -549,8 +549,8 @@ def test_debugprint_mitmot():
│ │ │ ├─ Subtensor{::step} [id BH] │ │ │ ├─ Subtensor{::step} [id BH]
│ │ │ │ ├─ Scan{scan_fn, while_loop=False, inplace=none} [id F] (outer_out_sit_sot-0) │ │ │ │ ├─ Scan{scan_fn, while_loop=False, inplace=none} [id F] (outer_out_sit_sot-0)
│ │ │ │ │ └─ ··· │ │ │ │ │ └─ ···
│ │ │ │ └─ ScalarConstant{-1} [id BI] │ │ │ │ └─ -1 [id BI]
│ │ │ └─ ScalarConstant{-1} [id BJ] │ │ │ └─ -1 [id BJ]
│ │ └─ ScalarFromTensor [id BK] │ │ └─ ScalarFromTensor [id BK]
│ │ └─ Sub [id C] │ │ └─ Sub [id C]
│ │ └─ ··· │ │ └─ ···
...@@ -560,41 +560,41 @@ def test_debugprint_mitmot(): ...@@ -560,41 +560,41 @@ def test_debugprint_mitmot():
│ │ │ │ ├─ Scan{scan_fn, while_loop=False, inplace=none} [id F] (outer_out_sit_sot-0) │ │ │ │ ├─ Scan{scan_fn, while_loop=False, inplace=none} [id F] (outer_out_sit_sot-0)
│ │ │ │ │ └─ ··· │ │ │ │ │ └─ ···
│ │ │ │ └─ ExpandDims{axes=[0, 1]} [id BO] │ │ │ │ └─ ExpandDims{axes=[0, 1]} [id BO]
│ │ │ │ └─ TensorConstant{0.0} [id BP] │ │ │ │ └─ 0.0 [id BP]
│ │ │ ├─ IncSubtensor{i} [id BQ] │ │ │ ├─ IncSubtensor{i} [id BQ]
│ │ │ │ ├─ Second [id BR] │ │ │ │ ├─ Second [id BR]
│ │ │ │ │ ├─ Subtensor{start:} [id BS] │ │ │ │ │ ├─ Subtensor{start:} [id BS]
│ │ │ │ │ │ ├─ Scan{scan_fn, while_loop=False, inplace=none} [id F] (outer_out_sit_sot-0) │ │ │ │ │ │ ├─ Scan{scan_fn, while_loop=False, inplace=none} [id F] (outer_out_sit_sot-0)
│ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ └─ ···
│ │ │ │ │ │ └─ ScalarConstant{1} [id BT] │ │ │ │ │ │ └─ 1 [id BT]
│ │ │ │ │ └─ ExpandDims{axes=[0, 1]} [id BU] │ │ │ │ │ └─ ExpandDims{axes=[0, 1]} [id BU]
│ │ │ │ │ └─ TensorConstant{0.0} [id BV] │ │ │ │ │ └─ 0.0 [id BV]
│ │ │ │ ├─ Second [id BW] │ │ │ │ ├─ Second [id BW]
│ │ │ │ │ ├─ Subtensor{i} [id BX] │ │ │ │ │ ├─ Subtensor{i} [id BX]
│ │ │ │ │ │ ├─ Subtensor{start:} [id BS] │ │ │ │ │ │ ├─ Subtensor{start:} [id BS]
│ │ │ │ │ │ │ └─ ··· │ │ │ │ │ │ │ └─ ···
│ │ │ │ │ │ └─ ScalarConstant{-1} [id BY] │ │ │ │ │ │ └─ -1 [id BY]
│ │ │ │ │ └─ ExpandDims{axis=0} [id BZ] │ │ │ │ │ └─ ExpandDims{axis=0} [id BZ]
│ │ │ │ │ └─ Second [id CA] │ │ │ │ │ └─ Second [id CA]
│ │ │ │ │ ├─ Sum{axes=None} [id CB] │ │ │ │ │ ├─ Sum{axes=None} [id CB]
│ │ │ │ │ │ └─ Subtensor{i} [id BX] │ │ │ │ │ │ └─ Subtensor{i} [id BX]
│ │ │ │ │ │ └─ ··· │ │ │ │ │ │ └─ ···
│ │ │ │ │ └─ TensorConstant{1.0} [id CC] │ │ │ │ │ └─ 1.0 [id CC]
│ │ │ │ └─ ScalarConstant{-1} [id BY] │ │ │ │ └─ -1 [id BY]
│ │ │ └─ ScalarConstant{1} [id BT] │ │ │ └─ 1 [id BT]
│ │ └─ ScalarConstant{-1} [id CD] │ │ └─ -1 [id CD]
│ ├─ Alloc [id CE] (outer_in_sit_sot-0) │ ├─ Alloc [id CE] (outer_in_sit_sot-0)
│ │ ├─ TensorConstant{0.0} [id CF] │ │ ├─ 0.0 [id CF]
│ │ ├─ Add [id CG] │ │ ├─ Add [id CG]
│ │ │ ├─ Sub [id C] │ │ │ ├─ Sub [id C]
│ │ │ │ └─ ··· │ │ │ │ └─ ···
│ │ │ └─ TensorConstant{1} [id CH] │ │ │ └─ 1 [id CH]
│ │ └─ Subtensor{i} [id CI] │ │ └─ Subtensor{i} [id CI]
│ │ ├─ Shape [id CJ] │ │ ├─ Shape [id CJ]
│ │ │ └─ A [id P] │ │ │ └─ A [id P]
│ │ └─ ScalarConstant{0} [id CK] │ │ └─ 0 [id CK]
│ └─ A [id P] (outer_in_non_seqs-0) │ └─ A [id P] (outer_in_non_seqs-0)
└─ ScalarConstant{-1} [id CL] └─ -1 [id CL]
Inner graphs: Inner graphs:
...@@ -642,28 +642,28 @@ def test_debugprint_compiled_fn(): ...@@ -642,28 +642,28 @@ def test_debugprint_compiled_fn():
out = pytensor.function([M], out, updates=updates, mode="FAST_RUN") out = pytensor.function([M], out, updates=updates, mode="FAST_RUN")
expected_output = """Scan{scan_fn, while_loop=False, inplace=all} [id A] 2 (outer_out_sit_sot-0) expected_output = """Scan{scan_fn, while_loop=False, inplace=all} [id A] 2 (outer_out_sit_sot-0)
├─ TensorConstant{20000} [id B] (n_steps) ├─ 20000 [id B] (n_steps)
├─ TensorConstant{[ 0 ... 998 19999]} [id C] (outer_in_seqs-0) ├─ [ 0 ... 998 19999] [id C] (outer_in_seqs-0)
├─ SetSubtensor{:stop} [id D] 1 (outer_in_sit_sot-0) ├─ SetSubtensor{:stop} [id D] 1 (outer_in_sit_sot-0)
│ ├─ AllocEmpty{dtype='int64'} [id E] 0 │ ├─ AllocEmpty{dtype='int64'} [id E] 0
│ │ └─ TensorConstant{20000} [id B] │ │ └─ 20000 [id B]
│ ├─ TensorConstant{(1,) of 0} [id F] │ ├─ [0] [id F]
│ └─ ScalarConstant{1} [id G] │ └─ 1 [id G]
└─ <Tensor3(float64, shape=(20000, 2, 2))> [id H] (outer_in_non_seqs-0) └─ <Tensor3(float64, shape=(20000, 2, 2))> [id H] (outer_in_non_seqs-0)
Inner graphs: Inner graphs:
Scan{scan_fn, while_loop=False, inplace=all} [id A] Scan{scan_fn, while_loop=False, inplace=all} [id A]
← Composite{switch(lt(i0, i1), i2, i0)} [id I] (inner_out_sit_sot-0) ← Composite{switch(lt(i0, i1), i2, i0)} [id I] (inner_out_sit_sot-0)
├─ TensorConstant{0} [id J] ├─ 0 [id J]
├─ Subtensor{i, j, k} [id K] ├─ Subtensor{i, j, k} [id K]
│ ├─ *2-<Tensor3(float64, shape=(20000, 2, 2))> [id L] -> [id H] (inner_in_non_seqs-0) │ ├─ *2-<Tensor3(float64, shape=(20000, 2, 2))> [id L] -> [id H] (inner_in_non_seqs-0)
│ ├─ ScalarFromTensor [id M] │ ├─ ScalarFromTensor [id M]
│ │ └─ *0-<Scalar(int64, shape=())> [id N] -> [id C] (inner_in_seqs-0) │ │ └─ *0-<Scalar(int64, shape=())> [id N] -> [id C] (inner_in_seqs-0)
│ ├─ ScalarFromTensor [id O] │ ├─ ScalarFromTensor [id O]
│ │ └─ *1-<Scalar(int64, shape=())> [id P] -> [id D] (inner_in_sit_sot-0) │ │ └─ *1-<Scalar(int64, shape=())> [id P] -> [id D] (inner_in_sit_sot-0)
│ └─ ScalarConstant{0} [id Q] │ └─ 0 [id Q]
└─ TensorConstant{1} [id R] └─ 1 [id R]
Composite{switch(lt(i0, i1), i2, i0)} [id I] Composite{switch(lt(i0, i1), i2, i0)} [id I]
← Switch [id S] 'o0' ← Switch [id S] 'o0'
......
tests/tensor/test_var.py
浏览文件 @ 1944353c
...@@ -213,7 +213,7 @@ def test_print_constant(): ...@@ -213,7 +213,7 @@ def test_print_constant():
c = pytensor.tensor.constant(1, name="const") c = pytensor.tensor.constant(1, name="const")
assert str(c) == "const{1}" assert str(c) == "const{1}"
d = pytensor.tensor.constant(1) d = pytensor.tensor.constant(1)
assert str(d) == "TensorConstant{1}" assert str(d) == "1"
@pytest.mark.parametrize( @pytest.mark.parametrize(
......
tests/test_printing.py
浏览文件 @ 1944353c
...@@ -280,10 +280,10 @@ def test_debugprint(): ...@@ -280,10 +280,10 @@ def test_debugprint():
│ ├─ AllocEmpty{dtype='float64'} 1 │ ├─ AllocEmpty{dtype='float64'} 1
│ │ └─ Shape_i{0} 0 │ │ └─ Shape_i{0} 0
│ │ └─ B │ │ └─ B
│ ├─ TensorConstant{1.0} │ ├─ 1.0
│ ├─ B │ ├─ B
│ ├─ <Vector(float64, shape=(?,))> │ ├─ <Vector(float64, shape=(?,))>
│ └─ TensorConstant{0.0} │ └─ 0.0
├─ D ├─ D
└─ A └─ A
......
Markdown 格式
0% 或
您添加了 0 人 到此讨论。请谨慎行事。
请先完成此评论的编辑！
请 注册 或者 后发表评论