我正在比较 XLAxla::Map
和元素操作,我想弄清楚何时使用它们。
API
快速浏览一下,似乎所有对标量的计算也可以定义为任意形状,因为所有相关的操作都有逐元素的变体(Eq
、Sin
等Add
)。我不清楚Map
除了替代 API 还提供什么。或者只是为了让我们可以使用预先构建的标量计算,在这种情况下它与 有何不同Call
?
表现
是否xla::Map
具有与元素操作相同的性能特征?也就是说,在伪代码中,是
Map(x, Abs)
Map(x, compose(Neg, Abs))
Map({x, y}, Eq)
优化至少大致与
Abs(x)
Neg(Abs(x))
Eq(x, y)