码迷,mamicode.com
首页 > 其他好文 > 详细

tensorflow之tf.meshgrid()

时间:2019-11-08 22:34:40      阅读:157      评论:0      收藏:0      [点我收藏+]

标签:分析   网格   span   mesh   style   size   矩阵   产生   结果   

tf.meshgrid()用于从数组a和b产生网格。生成的网格矩阵A和B大小是相同的,它也可以是更高维的。

用法:[A, B] = Meshgrid(a, b), 生成size(b) x size(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]]

结果分析:size(b)=5, 所以产生5行a,; size(a)=3, b先转置成一列,然后增加到3行。 

tensorflow之tf.meshgrid()

标签:分析   网格   span   mesh   style   size   矩阵   产生   结果   

原文地址:https://www.cnblogs.com/elitphil/p/11823631.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!