首页 > 其他 > 详细

tensorflow之tf.meshgrid()

时间:2019-11-09 01:10:34      阅读:382      评论:0      收藏:0      [点我收藏+]

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()

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

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!