Python 如何获得tensorflow keras中Lambda层内部两个张量（形状=[？，256512]）的成对欧氏距离？

我有两个形状的张量（？，256512）。我的目标是计算沿轴=2（第三轴）的成对欧几里德距离。我试过这个解决办法。在这个解决方案中，他们使用了已知形状的二维张量（例如：（2,3））。然而，在我的例子中，我尝试使用（？，256512）的输入形状构建一个自定义层，其中？=批次号，256=特征号，512=特征深度。预期的输出形状为（256256）

我面临的问题是由于在构建模型时所提到的形状中的“？”造成的。我正在寻找一个解决方案，可以实现具体使用tensorflow