numpy - 将 numpy 转换为张量（keras）（损失函数）

Question

我正在尝试在以下形状上定义损失函数： (NUM_OF_STROKES, STROKE_LEN, 2) 例如，假设 NUM_OF_STROKES=1, STROKE_LEN=4，它可以像： [[[x1,y1], [x2,y2], [x3,y3], [x4,y4]]]

我希望我的损失函数是两个对应点之间的距离（所有距离的总和）。例如：

p1 = [[[a1,b1], [a2,b2], [a3,b3], [a4,b4]]]
p2 = [[[c1,d1], [c2,d2], [c3,d3], [c4,d4]]]
loss = sqrt((a1-c1)^2 + (b1-d1)^2) + ... + sqrt((a4-c4)^2 + (b4-d4)^2)

在 numpy 中，我可以这样做： np.sum(np.linalg.norm(np.array(p1) - np.array(p2), axis=1))

但我不知道如何在 tensrflow 中做到这一点，我正在使用 tensorflow 2 和 keras。

标签： numpykerastensorflow2.0tensorloss-function

Answer 1

我认为您正在寻找的是：

tf.keras.backend.sum(tf.sqrt(tf.keras.backend.sum(tf.square(labels - predictions), axis=3)))

和axis = 3批次