Split operator in Relay?

Questions
#1

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?

Thanks.

1 Like
#2

Hello! I’m also interested in this topic.

1 Like
#3

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?

#5

I’m looking for something similar, did you find any reference?