提交 f4a78f95 authored 作者: Pascal Lamblin's avatar Pascal Lamblin

Add test case

上级 59b58be8
......@@ -234,3 +234,14 @@ def test_gemv_zeros():
tmp = f(A, b)
assert np.allclose(tmp,
np.zeros((dim,)))
def test_gemv_dot_strides():
# Reported in https://github.com/Theano/Theano/issues/6142
xv = rand(5)
yv = rand(5, 1)
x = gpuarray_shared_constructor(xv)
y = gpuarray_shared_constructor(yv, broadcastable=(False, True))
f = theano.function([], tensor.dot(x, y[::-1]), mode=mode_with_gpu)
out = f()
utt.assert_allclose(out, np.dot(xv, yv[::-1]))
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论