tensor - Adding together results from multiple workers with ignite - Stack Overflow

admin管理员组

文章数量:1387409

I'm trying to figure out if ignite is a good fit for my application.

I have a large collection of records which will be processed by multiple workers running in separate processes.

Each worker will generate several tensors in different streams, e.g., Tensor1, Tensor2, ... The entries for each tensor will always have the same size/shape.

What I would like Ignite to do is to add together all of the tensors in the Tensor1 stream, the Tensor2 stream, &c so when I'm done, I'll be able to fetch the sum from each stream.

From a quick look at the documentation, I think that Ignite might support this, but is there some other platform I should consider instead?

Thanks for any guidance you can give.

本文标签： tensorAdding together results from multiple workers with igniteStack Overflow