Tensor Operations

xxxxxxxxxx

 

1

import numpy

2

A=numpy.array(range(1,9)).reshape(2,2,2)

3

B=['A','B','C','D']

4

B=numpy.array(B,dtype=object).reshape(2,2)

5

AB1=numpy.tensordot(A,B,1)

6

AB01=numpy.tensordot(A,B,(0,1))

7

AB10=numpy.tensordot(A,B,(1,0))

8

def pp(string,color):

9

    start='<text style="color:'+color

10

    start+=';">${\\mathbb{'

11

    end='}}$</text>'

12

    return display(html(start+string+end)) 

13

def result(X):

14

    pretty_print(X.shape)

15

    for i in range(2): 

16

        for j in range(2): 

17

            for k in range(2):

18

                pp(X[i,j,k],'darkblue')

19

result(AB1); result(AB01); result(AB10)

Language: Sage Gap GP HTML Macaulay2 Maxima Octave Python R Singular

---

Messages