我正在尝试训练一个深度学习模型来预测一个 8*2 矩阵。预测矩阵将具有复数值，输入矩阵将是实数。可以做到吗？感谢您的时间。
如何训练深度学习模型来预测矩阵？
人工智能
深度学习
卷积神经网络
2021-11-09 12:14:24
1个回答
您可以使用 CNN 或全连接网络并输出大小为 8*2*2 的矩阵。第一个 8*2 矩阵是实数，第二个是虚数。下面的示例代码使用 keras。
'''
input: numpy array of shape(batch_size,input_dim_0,input_dim_1,1)
Y: keras tensor of shape(batch_size,8,2,2)
'''
model = Sequential()
model.add(Conv2d(64,3,input_dim =(input_dim_0,input_dim_1,1))
model.add(Conv2d(64,3)
model.add(Conv2d(2,3)
model.compile(loss='MSE',metrixs=['accuracy'])
model.fit(input,Y)
其它你可能感兴趣的问题