首页 > 其他 > 详细

t[..., 1, tf.newaxis]

时间:2020-04-21 16:20:54      阅读:82      评论:0      收藏:0      [点我收藏+]

如果t是二维数组,t[...,1]等价于t[:,1];如果是三维数值,t[...,1]等价于t[:,:,1]

tf.newaxisnp.newaxis功能相同,都是增加维度。

t=tf.constant([[1,2,3],[4,5,6]])
<tf.Tensor: id=25, shape=(2, 3), dtype=int32, numpy=
array([[1, 2, 3],
       [4, 5, 6]], dtype=int32)>
t[...,1]
<tf.Tensor: id=29, shape=(2,), dtype=int32, numpy=array([2, 5], dtype=int32)>
t[...,1,tf.newaxis]
<tf.Tensor: id=24, shape=(2, 1), dtype=int32, numpy=array([[2],[5]], dtype=int32)>

t[..., 1, tf.newaxis]

原文:https://www.cnblogs.com/yaos/p/12744929.html

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