在张量列表中查找最大绝对标量值

2024-10-01 19:31:37 发布

您现在位置:Python中文网/ 问答频道 /正文

我有一个不同维度的张量列表。我想找出所有张量的最大绝对标量值。问题是,我想我需要一种方法来做大于或小于张量,但我只能找到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数组,那么这将是微不足道的,但不幸的是,我需要在构建过程中这样做


Tags: 方法reduce列表iftfabsequal事情

热门问题