當前位置: 首頁>>代碼示例 >>用法及示例精選 >>正文


Python tf.math.reduce_euclidean_norm用法及代碼示例


計算張量維度上元素的歐幾裏得範數。

用法

tf.math.reduce_euclidean_norm(
    input_tensor, axis=None, keepdims=False, name=None
)

參數

  • input_tensor 要減少的張量。應該是數字類型。
  • axis 要減小的尺寸。如果None(默認),減少所有維度。必須在 [-rank(input_tensor), rank(input_tensor)) 範圍內。
  • keepdims 如果為真,則保留長度為 1 的縮減維度。
  • name 操作的名稱(可選)。

返回

  • 與 input_tensor 具有相同 dtype 的縮減張量。

沿 axis 中給定的尺寸減少 input_tensor 。除非 keepdims 為真,否則對於 axis 中的每個條目,張量的秩都會減少 1,這必須是唯一的。如果 keepdims 為真,則保留縮減後的維度,長度為 1。

如果axis 為None,則所有維度都會減少,並返回具有單個元素的張量。

例如:

x = tf.constant([[1, 2, 3], [1, 1, 1]]) # x.dtype is tf.int32
tf.math.reduce_euclidean_norm(x)  # returns 4 as dtype is tf.int32
y = tf.constant([[1, 2, 3], [1, 1, 1]], dtype = tf.float32)
tf.math.reduce_euclidean_norm(y)  # returns 4.1231055 which is sqrt(17)
tf.math.reduce_euclidean_norm(y, 0)  # [sqrt(2), sqrt(5), sqrt(10)]
tf.math.reduce_euclidean_norm(y, 1)  # [sqrt(14), sqrt(3)]
tf.math.reduce_euclidean_norm(y, 1, keepdims=True)  # [[sqrt(14)], [sqrt(3)]]
tf.math.reduce_euclidean_norm(y, [0, 1])  # sqrt(17)

相關用法


注:本文由純淨天空篩選整理自tensorflow.org大神的英文原創作品 tf.math.reduce_euclidean_norm。非經特殊聲明,原始代碼版權歸原作者所有,本譯文未經允許或授權,請勿轉載或複製。