|
|
1
1
你能提供更多关于你定义的层的细节吗? 无论如何,对于连接层,在同一轴上的两个张量之间必须至少有一个维度匹配。 简单来说,你可以很容易地看到“无”与1不匹配,16与64不匹配。 除此之外,尝试像你所做的那样重塑是错误的。最初h_t为形状(1,64),并具有:
您试图将它重新整形为(none,64)张量,这在概念上是错误的,因为none意味着它可以是任何整数,通常表示您的批大小。 尝试执行以下操作时出现相同问题:
当你试图把它重塑成一个固定的形状时,你的初始形状是(无,16):(1,16)。 我认为你应该试着让你的h-u层返回一个形状为(none,1,16)或(none,16)的张量,这样你就可以先展平,然后在第一种情况下连接,在第二种情况下只使用concatenate。 |
|
|
DPM · 将自定义掩码传递给LSTM数据以进行培训和验证 4 年前 |
|
|
user836026 · 在pytorch中复制权重时语法无效 4 年前 |
|
|
mad · 如何从一个定制的神经网络模型中获得逻辑和概率 4 年前 |