1import numpy
2from algopy import UTPM, dot
3
4D,P,M,N = 2,1,3,4
5x_data = 7*numpy.arange(D*P*M*N,dtype=float).reshape((D,P,M,N))
6y_data = numpy.arange(D*P*M*N,dtype=float).reshape((D,P,N,M))
7
8# calling algorithms directly
9z_data = numpy.zeros((D,P,M,M))
10UTPM._dot(x_data, y_data, z_data)
11
12# use UTPM instance
13x = UTPM(x_data)
14y = UTPM(y_data)
15z = dot(x, y)
16
17print('z.data - z_data', z.data - z_data)
18