提交 025fb494 authored 作者: Frederic Bastien's avatar Frederic Bastien

added missing test for tensor.{ceil,floor,iround}

上级 eb4d03d8
......@@ -385,6 +385,8 @@ PowInplaceTester = makeBroadcastTester(op = inplace.pow_inplace,
_good_broadcast_unary_normal_float = dict(normal = (rand_ranged(-5, 5, (2, 3)),))
_good_broadcast_unary_normal = dict(normal = (rand_ranged(-5, 5, (2, 3)),),
integers = (randint_ranged(-5, 5, (2, 3)),))
......@@ -418,6 +420,33 @@ SgnInplaceTester = makeBroadcastTester(op = inplace.sgn_inplace,
expected = numpy.sign,
good = _good_broadcast_unary_normal,
inplace = True)
CeilTester = makeBroadcastTester(op = ceil,
expected = lambda a: numpy.asarray(numpy.ceil(a),a.dtype),
good = _good_broadcast_unary_normal,
grad = _grad_broadcast_unary_normal)
CeilInplaceTester = makeBroadcastTester(op = inplace.ceil_inplace,
expected = lambda a: numpy.asarray(numpy.ceil(a),a.dtype),
good = _good_broadcast_unary_normal,
grad = _grad_broadcast_unary_normal,
inplace = True)
FloorTester = makeBroadcastTester(op = floor,
expected = lambda a: numpy.asarray(numpy.floor(a),a.dtype),
good = _good_broadcast_unary_normal,
grad = _grad_broadcast_unary_normal)
FloorInplaceTester = makeBroadcastTester(op = inplace.floor_inplace,
expected = lambda a: numpy.asarray(numpy.floor(a),a.dtype),
good = _good_broadcast_unary_normal,
grad = _grad_broadcast_unary_normal,
inplace = True)
IRoundTester = makeBroadcastTester(op = iround,
expected = lambda a: numpy.asarray(numpy.round(a),dtype='int64'),
good = _good_broadcast_unary_normal)
IRoundInplaceTester = makeBroadcastTester(op = inplace.iround_inplace,
expected = lambda a: numpy.asarray(numpy.round(a),dtype='int64'),
good = _good_broadcast_unary_normal,
inplace = True)
SqrTester = makeBroadcastTester(op = sqr,
expected = numpy.square,
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论