It also fails for

T = random.random((2,d,d))
W = random.random((2,d,d,i))

and d > 2. For d = 3 it fails for i = 911...1365.

Should I submit this as a bug (if so, how do I do that?) and/or contact the author Mark Wiebe?

Wieland

PS: How do I reply directly to your messages?