张量流为批量梯度洗牌张量

埃尔文·庞 |

敬启者,

我对张量流很陌生。我正在尝试为 CNN 解决著名的 MNIST 问题。但是当我必须重新调整 x_training 数据(这是一个 [40000, 28, 28, 1] 形状数据时遇到了困难。

我的代码如下:

x_train_final = tf.reshape(x_train_final, [-1, image_width, image_width, 1])

x_train_final = tf.cast(x_train_final, dtype=tf.float32)

perm = np.arange(num_training_example).astype(np.int32)
np.random.shuffle(perm)

x_train_final = x_train_final[perm]

发生以下错误:

ValueError: Shape must be rank 1 but is rank 2 for 'strided_slice_1371' (op: 'StridedSlice') with input shape: [40000,28,28,1], [1,40000], [1,40000], [1 ]。

任何人都可以建议我如何解决这个问题?谢谢。

普拉萨德

我建议您使用 scikit 的shuffle功能。

from sklearn.utils import shuffle
x_train_final = shuffle(x_train_final)

此外,您可以传入多个数组,shuffle函数将重新组织(混洗)这些多个数组中的数据,并在所有这些数组中保持相同的混洗顺序。因此,您甚至可以传入您的标签数据集。前任:

X_train, y_train = shuffle(X_train, y_train)

本文收集自互联网,转载请注明来源。

如有侵权,请联系[email protected] 删除。

编辑于
0

我来说两句

0条评论
登录后参与评论

相关文章

来自分类Dev

梯度值张量流的停止条件

来自分类Dev

用张量流梯度带计算Hessian

来自分类Dev

张量流中未聚合的梯度/每个示例的梯度

来自分类Dev

在张量流中停止带有 n/a 标签的梯度

来自分类Dev

如何在张量流中停止 LSTMStateTuple 的梯度

来自分类Dev

置换张量流张量

来自分类Dev

在张量流中扩展张量

来自分类Dev

使用张量流实现批量归一化

来自分类Dev

张量流中的批量归一化:变量和性能

来自分类Dev

在张量流中使用标量进行批量乘法/除法

来自分类Dev

复制张量流图

来自分类Dev

重载张量流模型

来自分类Dev

张量流懒吗?

来自分类Dev

张量流中的外积

来自分类Dev

张量流重塑用法

来自分类Dev

重载张量流模型

来自分类Dev

Python 3.6 张量流

来自分类Dev

张量流 RNN 实现

来自分类Dev

张量流,删除重复

来自分类Dev

图像排序张量流

来自分类Dev

InvalidArgumentError 张量流

来自分类Dev

如何在张量流中实现多元线性随机梯度下降算法?

来自分类Dev

张量流中是否可以分别计算两种梯度

来自分类Dev

我可以在张量流的损失函数内将梯度转换为标量吗?

来自分类Dev

使用tf.GradientTape()同时计算张量流中多层的损耗并计算梯度

来自分类Dev

如何对张量流中的并非所有变量进行梯度下降

来自分类Dev

在张量流中使用梯度下降优化器时获取变量的“nan”

来自分类Dev

张量流中张量列表的总和

来自分类Dev

在张量流中更改张量的比例