May 2017
Intermediate to advanced
270 pages
6h 18m
English
In this subsection, we will demonstrate the usage of a GPU with Theano and Tensorflow. As an example, we will benchmark the execution of a very simple matrix multiplication on the GPU and compare it to its running time on a CPU.
The following code performs a simple matrix multiplication using Theano. We use the T.matrix function to initialize a two-dimensional array, and then we use the T.dot method to perform the matrix multiplication:
from theano import function, config import theano.tensor as T import numpy as np import time ...Read now
Unlock full access