(mnist_images, mnist_labels), _ = tf.keras.datasets.fashion_mnist.load_data()
mnist_images.shape
# 结果如下
(60000, 28, 28)
tf.cast(x=mnist_images[...,**tf.newaxis**],dtype=tf.float32).shape
# 结果如下
TensorShape([60000, 28, 28, 1])
给Tensor添加新的轴
最新推荐文章于 2023-03-18 15:45:07 发布