2017-02-10 4 views
1

В SymPy, я бы что-то вроде:Как получить список переменных, от которых зависит выражение anano?

In [6]: import sympy as sp 

In [7]: sp.var('x, y') 
Out[7]: (x, y) 

In [8]: X = x + y 

In [9]: X.free_symbols 
Out[9]: {y, x} 

, чтобы получить переменные X зависит. Это супер удобно, потому что, если мы хотим сделать lambdify потом:

f = sp.lambdify(tuple(X.free_symbols), X) 

Я хотел бы сделать что-то подобное с Феано:

import theano 
import theano.tensor as T 
x, y = T.dvectors('x', 'y') 
X = x + y 
f = theano.function([x, y ], X) 

Но, вместо того, чтобы обеспечить [x,y] Я хотел бы получить доступ непосредственно список переменных, необходимых для создания theano.function

Возможно ли это? Если это так, я не нашел его в документе theano, поэтому любая помощь или ссылка были бы оценены :)

ответ

2

Некоторые функции anano не очень хорошо документированы, поскольку в основном для внутреннего использования.

import theano 
import theano.tensor as T 
x, y = T.vectors('xy') 
z = x+y 
theano.gof.graph.inputs([z]) 

Выход:

[x, y] 
Смежные вопросы