tf.newaxis

狼啸风云

修改于 2022-09-02 20:48:45

1.2K0

修改于 2022-09-02 20:48:45

给tensor增加维度

import tensorflow as tf

a = tf.range(10,dtype=tf.int32)
print(a)
b = a[:,tf.newaxis]
print(b)

a的shape为（10，） b的shape为（10，1）若b = a[tf.newaxis,:]，则b的shape为（1,10）

本文参与腾讯云自媒体同步曝光计划，分享自作者个人站点/博客。

原始发表：2020/08/31 ，如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自作者个人站点/博客前往查看

如有侵权，请联系 cloudcommunity@tencent.com 删除。

本文参与腾讯云自媒体同步曝光计划，欢迎热爱写作的你一起参与！

登录后参与评论

0 条评论

热度