tf.norm錯誤ValueError異常：「奧德」必須是一個支持向量範數，得到了來回

Question

我試圖計算我張的Frobenius範

W = tf.Variable(tf.random_normal([3072,20],stddev=0.1)) 
temp = tf.matmul(tf.transpose(W),W) 
fro_W = tf.norm(temp, ord ='fro') 

這將產生以下錯誤：

ValueError異常：'ord'必須是一個支持的向量規範，來回

我不明白爲什麼它把我的2D張量當作一個向量而不是矩陣。

我失去了一些東西在這裏？

謝謝

Answer 1

從documentation：

The Frobenius norm fro is not defined for vectors

此外，

If axis is None (the default), the input is considered a vector

試試這個：

tf.norm(temp, ord='fro', axis=(0,1))