def test_view(): for dtype in ['float16', 'float32']: a = rand_gpuarray(20, dtype=dtype) g = GpuArrayType(dtype=dtype, broadcastable=(False, ))('g') f = theano.function([g], ViewOp()(g)) assert isinstance(f.maker.fgraph.toposort()[0].op, ViewOp) res = f(a) assert GpuArrayType.values_eq(res, a)
def test_view(): for dtype in ["float16", "float32"]: a = rand_gpuarray(20, dtype=dtype) g = GpuArrayType(dtype=dtype, broadcastable=(False, ))("g") m = theano.compile.get_default_mode().excluding("local_view_op") f = theano.function([g], ViewOp()(g), mode=m) assert isinstance(f.maker.fgraph.toposort()[0].op, ViewOp) res = f(a) assert GpuArrayType.values_eq(res, a)