ListAddΒΆ
Functional interface to the Add layer (elementwise sum). Takes a list of tensors.
Abstract Signature:
ListAdd(inputs: List[Tensor])
PyTorch
API:
βStrategy: Plugin (reduce_sum_list)
JAX (Core)
API:
sumStrategy: Plugin (reduce_sum_list)