tensorflow2.0 -- 19 高阶OP

最新推荐文章于 2024-10-04 13:52:58 发布

林钟即望

最新推荐文章于 2024-10-04 13:52:58 发布

阅读量361

点赞数

分类专栏： tensorflow2 文章标签： tensorflow scatter mesh

本文链接：https://blog.csdn.net/weixin_45177485/article/details/112259869

版权

本文详细介绍了TensorFlow 2.0中的高阶操作，包括：1) Where操作，讲解了where(mask)和where(cond,A,B)的用法；2) Scatter Nd，阐述如何按照坐标更新矩阵；3) Meshgrid，解释了如何创建坐标网络并利用GPU进行运算。通过这些操作，可以更高效地处理TensorFlow中的数据和条件逻辑。" 123893491,5564205,大数据环境下的CountDistinct去重技术解析,"['大数据开发', 'Spark', 'Hive', 'Kylin']

摘要由CSDN通过智能技术生成

高阶OP

高阶操作

高阶操作

1 where

where(mask)

筛选出矩阵中true位置的坐标：

>>> a = tf.random.normal([3,3])
<tf.Tensor: shape=(3, 3), dtype=float32, numpy=
array([[-0.9279645 ,  0.04238617,  0.1136281 ],
       [-0.91321355,  0.96097076, -0.9072119 ],
       [ 0.699401  , -0.39297295,  0.73130745]], dtype=float32)>
>>> mask = a>0
<tf.Tensor: shape=(3, 3), dtype=bool, numpy=
array([[False,  True,  True],
       [False,  True, False],
       [ True, False,  True]])>
>>> indices = tf.where(mask)
<tf.Tensor: shape=(5, 2), dtype=int64, numpy=
array([[0, 1],
       [0, 2],
       [1, 1],
       [2