nn.Add works with batches