当前位置: 首页 > 知识库问答 >
问题:

如何解决此类型错误:只能将整数标量数组转换为标量索引

卢毅
2023-03-14

我使用np。随机的选择获取随机索引,但遇到以下错误。

TypeError:只能将整数标量数组转换为标量索引。任何建议都将不胜感激。

x_train = [0,1,2,3,4]
train_size = 4
batch_size =3
batch_mask = np.random.choice(train_size, batch_size)
print(batch_mask)
x_batch = x_train[batch_mask]
print(x_batch)

共有1个答案

郭意
2023-03-14

Xu train是一个简单的python列表。您试图使用NumPy数组作为索引;这是不合法的。将x_列改为numpy数组。

x_train = np.array([0,1,2,3,4])
train_size = 4
batch_size =3
batch_mask = np.random.choice(train_size, batch_size)
print(batch_mask)
x_batch = x_train[batch_mask]
print(x_batch)

输出:

[3 3 2]
[13 13 12]
 类似资料: