簡體   English   中英

在張量列表中找到最大絕對標量值

[英]Find maximum absolute scalar value in list of tensors

我有張量的列表,尺寸不同。 我想找到所有張量的最大絕對標量值。 麻煩的是,我想我需要一種大於或小於張量的方法,但是我只能找到tf.equal() 這是我想做的事情:

curMaxAbs = tf.Variable(-1, tf.float64)
for g in myList:
    maxG = tf.abs(tf.reduce_max(g))
    minG = tf.abs(tf.reduce_min(g))
    maxAbsG = maxG if tf.greaterThan(maxG,minG) else minG
    curMaxAbs = maxAbsG if tf.greaterThan(maxAbsG, curMaxAbs) else curMaxAbs

當然,似乎沒有tf.greaterThan()函數。 顯然,如果我可以使用tf.eval()並轉換為numpy數組,這將是微不足道的,但是不幸的是,我需要在構造過程中執行此操作。

像這樣的tf.maximum怎么樣:

maxAbsG = tf.maximum ( maxG, minG )

暫無
暫無

聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM