tensorflow之tf.meshgrid()

meshgrid用於從數組a和b產生網格。生成的網格矩陣A和B大小是相同的。它也可以是更高維的。用法: [A,B]=Meshgrid(a,b)，生成size(b)Xsize(a)大小的矩陣A和B。它相當於a從一行重復增加到size(b)行，把b轉置成一列再重復增加到size(a)列

a=[0,5,10]
b=[0,5,15,20,25]
A,B=tf.meshgrid(a,b)
with tf.Session() as sess:
print (A.eval())
print (B.eval())

結果：
[[ 0 5 10]
[ 0 5 10]
[ 0 5 10]
[ 0 5 10]
[ 0 5 10]]
[[ 0 0 0]
[ 5 5 5]
[15 15 15]
[20 20 20]
[25 25 25]]

舉例，含有三個參數

 

參考文獻：https://blog.csdn.net/mou_it/article/details/82083984