提交 ba70cb82 authored 作者: Ying Zhang's avatar Ying Zhang

add basic test

上级 5716f992
import numpy
import theano
import theano.tensor as T
from theano.tensor.opt import Assert
def test_assert_op_gradient():
x = T.vector('x')
assert_op = Assert()
cost = T.sum(assert_op(x, x.size<2))
grad = T.grad(cost, x)
func = theano.function([x], grad)
x_val = numpy.ones(shape=(1,))
assert func(x_val) == 1
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论