def test_simple(self): x = matrix("x") y = x + x fgraph = FunctionGraph([x], [y], clone=False) shapes = shape_of_variables(fgraph, {x: (5, 5)}) assert shapes == {x: (5, 5), y: (5, 5)} x = matrix("x") y = aet.dot(x, x.T) fgraph = FunctionGraph([x], [y], clone=False) shapes = shape_of_variables(fgraph, {x: (5, 1)}) assert shapes[x] == (5, 1) assert shapes[y] == (5, 5)
def test_err(self): x = matrix("x") subx = x[1:] fgraph = FunctionGraph([x], [subx]) with pytest.raises(ValueError): shape_of_variables(fgraph, {x: (10, 10)})
def test_subtensor(self): x = matrix("x") subx = x[1:] fgraph = FunctionGraph([x], [subx], clone=False) shapes = shape_of_variables(fgraph, {x: (10, 10)}) assert shapes[subx] == (9, 10)