tf.newaxis

时间:2022-07-24

本文章向大家介绍tf.newaxis，主要内容包括其使用实例、应用技巧、基本知识点总结和需要注意事项，具有一定的参考价值，需要的朋友可以参考一下。

QQ：战、血舞皇朝、2125364717

给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）

各位看官老爷，如果觉得对您有用麻烦赏个子，创作不易，0.1元就行了。下面是微信乞讨码：

添加描述

随机文章

本站知识点必读