Re: Performance of Array Addition
- To: mathgroup at smc.vnet.net
- Subject: [mg90982] Re: [mg90952] Performance of Array Addition
- From: DrMajorBob <drmajorbob at att.net>
- Date: Fri, 1 Aug 2008 03:02:16 -0400 (EDT)
- References: <12864392.1217497029287.JavaMail.root@m08>
- Reply-to: drmajorbob at longhorns.com
Here are timings of your methods and a couple of my own. First with integers, then with reals. Clear[integers] integers[d1_, d2_] := RandomInteger[10, {d1, d2, 3}] array = integers[3000, 4000]; Timing[one = array[[All, All, 1]] + array[[All, All, 2]] + array[[All, All, 3]];] Timing[two = Map[Total, array, {2}];] Timing[three = Plus @@ Transpose[array, {2, 3, 1}];] Timing[four = Total@Transpose[array, {2, 3, 1}];] one == two == three == four {1.06188, Null} {12.6128, Null} {0.623087, Null} {0.69472, Null} True (I test first with integers so that there's less ambiguity in the test for equality.) Clear[reals] reals[d1_, d2_] := RandomInteger[10, {d1, d2, 3}] array = reals[3000, 4000]; Timing[one = array[[All, All, 1]] + array[[All, All, 2]] + array[[All, All, 3]];] Timing[two = Map[Total, array, {2}];] Timing[three = Plus @@ Transpose[array, {2, 3, 1}];] Timing[four = Total@Transpose[array, {2, 3, 1}];] one == two == three == four {1.07193, Null} {12.323, Null} {0.624647, Null} {0.705247, Null} True > The advantage of Map[Total, array, {2}] is that it allows arrays of > dimension {d1,d2,d3}, with d3 arbitrary. I believe my methods also have that property. integers[d1_, d2_, d3_] := RandomInteger[10, {d1, d2, d3}] array = integers[3000, 400, 100]; Timing[two = Map[Total, array, {2}];] Timing[three = Plus @@ Transpose[array, {2, 3, 1}];] Timing[four = Total@Transpose[array, {2, 3, 1}];] two == three {2.13325, Null} {2.18372, Null} {1.98925, Null} True array = integers[300, 4000, 100]; Timing[two = Map[Total, array, {2}];] Timing[three = Plus @@ Transpose[array, {2, 3, 1}];] Timing[four = Total@Transpose[array, {2, 3, 1}];] two == three {2.13839, Null} {2.18131, Null} {1.97734, Null} True > Is there a way to get closer to the speed advantages of hand-coding > the addition of the final dimension, while > allowing the number of elements in the final dimension to be > arbitrary? On average, I suspect Total@Transpose is competitive. Bobby On Thu, 31 Jul 2008 01:57:38 -0500, Jeremy <jeremytfox at mac.com> wrote: > Let there be, for example, an array of reals with dimensions > {d1,d2,3}. > > I have noticed that the command > > array[[All,All,1]] + array[[All,All,2]] + array[[All,All,3]]; > > is much faster than > > Map[Total, array, {2}]; > > even though they return the same answer. > > The advantage of Map[Total, array, {2}] is that it allows arrays of > dimension {d1,d2,d3}, with d3 arbitrary. > > Is there a way to get closer to the speed advantages of hand-coding > the addition of the final dimension, while > allowing the number of elements in the final dimension to be > arbitrary? > > Thanks, Jeremy > > -- DrMajorBob at longhorns.com