如果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是二维数组,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)>

相关文章:

  • 2022-12-23
  • 2021-10-04
  • 2021-07-14
  • 2022-12-23
  • 2021-04-21
  • 2022-01-19
  • 2022-12-23
  • 2021-09-22
猜你喜欢
  • 2021-11-11
  • 2022-12-23
  • 2021-07-12
  • 2021-07-03
  • 2022-01-04
相关资源
相似解决方案