仅当使用docker时，Tensorflow错误

我有我的笔记本电脑使用TensorFlow工作正常的CNN代码。 但是，当我使用"gcr.io/tensorflow/tensorflow"并在容器中安装所需的包时，出现以下错误：

InvalidArgumentError（请参阅上面的回溯）：除1之外的所有维度必须匹配。 input1具有形状[100 1]，并且与形状[135 2560]的input0不匹配。 [[Node：gradients / net / concat_grad / ConcatOffset = ConcatOffset [N = 2，_device =“/ job：localhost / replica：0 / task：0 / cpu：0”]（gradients / net / concat_grad / mod，gradients / net / concat_grad / ShapeN，gradients / net / concat_grad / ShapeN：1）]]

如果没有docker，没有形状不匹配的问题。 我在某个地方读到这可能是有意义的记忆问题。 但是，即使我给我的充分的内存，docker和减less批量大小，我仍然得到错误。 有谁知道是什么问题？