层激活的正交正则化

Question

我想在密集层的输出激活（Y）上添加正交正则化。即min || Y'Y-I || _F

def regOrth(Y):
   m = K.batch_dot(K.transpose(Y), Y) - K.eye(Y.shape[2])
   return K.sqrt(K.sum(K.square(m)))

输出Y的形状为（batch_size，W，H），我希望每个（WxH）矩阵都具有正交列。 由于批量大小我不能使用K.transpose作为第一个昏暗。是否有类似于K.batch_dot的后端功能。

我该如何实现？