Skip to content

Instantly share code, notes, and snippets.

@certik
Created September 7, 2010 19:08
Show Gist options
  • Save certik/568879 to your computer and use it in GitHub Desktop.
Save certik/568879 to your computer and use it in GitHub Desktop.
from numpy import zeros, dot, diag, tensordot
from numpy.random import random
n = 6
m = 4
b = random((n, m))
d = random((m, m))
bdb = zeros(n, 'd')
for i in xrange(n):
bdb[i] = dot(b[i,:], dot(d, b[i, :]))
print bdb
bd = dot(b, d)
bdb = diag(dot(bd, b.T))
print bdb
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment