for me what you want to do is not clear
You've a (2,3) matrix size so what product do you want to perform?
Maybe such kind (without Tensorflow here)?
You've a (2,3) matrix size so what product do you want to perform?
Maybe such kind (without Tensorflow here)?
import numpy as np N = 3 V = (1./np.sqrt(2))*np.ones((2,N)) TP1 = V[:,0]*V[:,1]*V[:,2] TP2 = np.prod(V) # print("TP1 = {}".format(TP1)) print("TP2 = {}".format(TP2))