def local_gpureshape(node): op = node.op name = op.name if name: name = 'Gpu' + name res = GpuReshape(op.ndim, op.name) return res
def local_gpuflatten(node): op = node.op if op.outdim != 1: return None res = GpuReshape(op.outdim, None) o = res(node.inputs[0], theano.tensor.constant([-1])) return o
def local_gpuflatten(node): op = node.op shp = [] if op.outdim != 1: shp = [node.inputs[0].shape[i] for i in range(op.outdim - 1)] shp += [-1] res = GpuReshape(op.outdim, None) o = res(node.inputs[0], theano.tensor.as_tensor_variable(shp)) return o