def test_fill_grad(self):
# Fix bug reported at
# https://groups.google.com/d/topic/theano-users/nQshB8gUA6k/discussion
x = TensorType(config.floatX, [0, 1, 0])('x')
y = TensorType(config.floatX, [0, 1, 0])('y')
e = tensor.second(x, y)
theano.grad(e.sum(), y)
def test_fill_grad(self):
# Fix bug reported at
# https://groups.google.com/d/topic/theano-users/nQshB8gUA6k/discussion
x = TensorType(config.floatX, [0, 1, 0])('x')
y = TensorType(config.floatX, [0, 1, 0])('y')
e = tensor.second(x, y)
theano.grad(e.sum(), y)
def test_second():
a0 = tt.scalar("a0")
b = tt.scalar("b")
out = theano.scalar.basic.second(a0, b)
fgraph = theano.gof.FunctionGraph([a0, b], [out])
compare_jax_and_py(fgraph, [10.0, 5.0])
a1 = tt.vector("a1")
out = tt.second(a1, b)
fgraph = theano.gof.FunctionGraph([a1, b], [out])
compare_jax_and_py(fgraph,
[np.zeros([5], dtype=theano.config.floatX), 5.0])