计算张量内的和

Question

我正在使用Tensorflow设计神经网络。我有一个形状为G的张量[batch_size, C, 1 + x + y, feature_size]，我想计算x内y和G个向量的总和，以便新G 1}}具有形状：[batch_size, C, 3, feature_size]。那么我应该使用Tensorflow中的哪个操作？

感谢您的帮助。

Answer 1

尝试以下方法：

mylist
     .stream()
     .parallel() // for parallel processing
     .map(variant -> variant.getSku())