from singa import layer
from singa import net as ffnet
from singa import loss
from singa import device
from singa import tensor

# create the net
net = net = ffnet.FeedForwardNet(loss.SoftmaxCrossEntropy())
net.add(layer.Conv2D('conv', 32, 5, 1, pad=2, (3, 32, 32)))
net.add(layer.Activation('relu'))
net.add(layer.MaxPooling2D('pool', 3, 2))
net.add(layer.Flatten('flat'))
net.add(layer.Dense('dense', 10))

dev = device.create_cuda_gpu()
net.to_device(dev)

x = tensor.from_numpy(image)
x.to_device(dev)
y= net.predict(x)