gpt4 book ai didi

python - tensorflow map_fn 是否支持采用多个张量?

转载 作者:太空狗 更新时间:2023-10-29 18:20:55 28 4
gpt4 key购买 nike

tf.map_fn 是否像 python 的 native map 函数(下面提供的示例)所支持的那样支持采用多个张量?

a = [1,2,3,4]
b = [17,12,11,10]
print(map(lambda x,y:x+y, a,b)) # ==> [18, 14, 14, 14]

最佳答案

和今天一样,我看到 map_fn 得到了增强,可以使用两个张量,正如文档中所说的那样 - “elems:一个张量或(可能嵌套的)张量序列,每个张量都将沿着它们的第一个维度解包。嵌套的结果切片的序列将应用于 fn。”该示例(尽管以 numpy 形式给出)还表明它可以采用两个张量。我正在这里复制它。

elems = (np.array([1, 2, 3]), np.array([-1, 1, -1]))
alternate = map_fn(lambda x: x[0] * x[1], elems, dtype=tf.int64)
# alternate == [-1, 2, -3]

Source:

关于python - tensorflow map_fn 是否支持采用多个张量?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/37086098/

28 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com