如何在tensorflow中切片数组

作者: Coolkid
发布时间: 2024-10-17 08:25:55 (18天前)
转自：

2 条回复

0#
回复此人
757461156 | 2019-08-31 10-32

<div class =“post-text”itemprop =“text”> <P> 您可以使用与numpy相同的操作。 </p> <pre> <code> a = tf.constant([[1,2,3,4],[5,6,7,8],[7,8,9,0]]) a.shape </code> </pre> <P> 的形状 <code> a </code> 是 <code> Out[]:TensorShape([Dimension(3), Dimension(4)]) </code> </p> <P> 使用切片操作，获取第二维中的值： </p> <pre> <code> a[:, 2].eval() </code> </pre> <P> 而输出是 </p> <pre> <code> array([3, 7, 9]) </code> </pre> </DIV>

编辑

登录后才能参与评论