Split operator in Relay?

I have two or more large tensors and I do some kind of operation such as +.

I’d like to split this tensor dynamically into parts and do the additions in parallel using the partitions of the tensor. I am wondering If Relay supports such kind of operator?


Hello! I’m also interested in this topic.

1 Like

I know there is a split operator in relay for tensors, but I am wondering if there is a similar split operator for actual operators like add?

As you mentioned the split operator, did you mean the relay.op.split here?