tensorflow 中的 Frobenius 归一化实现

Frobenius normalization implementation in tensorflow

我是 tensorflow 的初学者,我想在张量上应用 Frobenius normalization 但是当我搜索时我没有在 tensorflow 中找到任何与之相关的函数我无法使用 tensorflow ops 实现它,我可以使用 numpy 操作来实现它,但是我如何仅使用 tensorflow ops 来实现它??

我在 python

中使用 numpy 实现
def Frobenius_Norm(tensor):
   x = np.power(tensor,2)
   x = np.sum(x) 
   x = np.sqrt(x)
   return x 
def frobenius_norm_tf(M):
    return tf.reduce_sum(M ** 2) ** 0.5