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