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

如果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)>
posted @ 2020-04-21 15:26  2021年的顺遂平安君  阅读(1280)  评论(0编辑  收藏  举报