如何创建 nd4j 3D 输入数据

如何解决如何创建 nd4j 3D 输入数据

我已经训练了一个 keras 模型

model = Sequential()
model.add(LSTM(input_shape=(50,48),units=100,return_sequences=True))
model.add(Dropout(0.2))
model.add(LSTM(units=50,return_sequences=False))
model.add(Dropout(0.2))
model.add(Dense(128,activation='relu'))
model.add(Dropout(0.5))
model.add(Dense(64,activation='relu'))
model.add(Dropout(0.5))
model.add(Dense(units=1,activation='sigmoid'))
# optimizer = keras.optimizers.Adam(lr=0.0001)
model.compile(loss=[binary_focal_loss(alpha=.972,gamma=3)],optimizer='adam',metrics=['accuracy'])
model.summary()
return model

当我创建一个 inputData 时，我得到一个错误：3D input expected to RNN layer expected，得到 2。 Indarray input = Nd4j.create(inputData,new int[]{50,48});

我试过 Indarray input = Nd4j.create(inputData,new int[]{1,50,48}); 又犯了一个错误接收到 size(1) = 50 的输入（输入数组形状 = [1,48]）； input.size(1) 必须匹配层 nIn 大小 (nIn = 48)

怎么了