示例#1
0
文件: Shape.cs 项目: omegahm/bohrium
        /// <summary>
        /// Generates shapes that are broadcast compatible based on their current shapes
        /// </summary>
        /// <param name="self">A shape.</param>
        /// <param name="other">Another shape.</param>
        /// <returns>A tupple with shapes are broadcast-compatible with respect to the original shapes and the combined shapes</returns>
        public static Tuple<Shape, Shape> ToBroadcastShapes(Shape self, Shape other)
        {
            long selfDims = self.Dimensions.LongLength;
            long otherDims = other.Dimensions.LongLength;

            long sharedDims = Math.Min(selfDims, otherDims);
            long resultDims = Math.Max(selfDims, otherDims);

            long[] resultDimensions = new long[resultDims];

            for (long i = 0; i < sharedDims; i++)
            {
                long sizeSelf = self.Dimensions[selfDims - i - 1].Length;
                long sizeOther = other.Dimensions[otherDims - i - 1].Length;

                if (sizeSelf == sizeOther)
                    resultDimensions[resultDims - i - 1] = sizeSelf;
                else
                {
                    if (sizeSelf != sizeOther && sizeOther != 1 && sizeSelf != 1)
                        throw new Exception("Dimension sizes do not match for broadcast");
                    resultDimensions[resultDims - i - 1] = Math.Max(sizeSelf, sizeOther);
                }
            }

            Shape largest = selfDims > otherDims ? self : other;

            for (long i = sharedDims; i < resultDims; i++)
                resultDimensions[resultDims - i - 1] = largest.Dimensions[resultDims - i - 1].Length;

            long[] selfStrides = new long[resultDims];
            long[] otherStrides = new long[resultDims];

            long diffSelf = resultDims - selfDims;
            long diffOther = resultDims - otherDims;

            //Now the shape has broadcast-able dimension size, so fix up the strides to go with it
            for (long i = 0; i < resultDims; i++)
            {
                if (i - diffSelf >= 0)
                {
                    if (self.Dimensions[i - diffSelf].Length == 1)
                        selfStrides[i] = 0;
                    else
                        selfStrides[i] = (resultDimensions[i] / self.Dimensions[i - diffSelf].Length) * self.Dimensions[i - diffSelf].Stride;
                }
                else
                    selfStrides[i] = 0;

                if (i - diffOther >= 0)
                {
                    if (other.Dimensions[i - diffOther].Length == 1)
                        otherStrides[i] = 0;
                    else
                        otherStrides[i] = (resultDimensions[i] / other.Dimensions[i - diffOther].Length) * other.Dimensions[i - diffOther].Stride;
                }
                else
                    otherStrides[i] = 0;
            }

            return new Tuple<Shape, Shape>(
                new Shape(resultDimensions, self.Offset, selfStrides),
                new Shape(resultDimensions, other.Offset, otherStrides)
            );
        }
示例#2
0
        public static void RunTests()
        {
            var an = Generate.Empty(2000);
            var bn = Generate.Ones(2000);

            var cn = Generate.Empty(2000);
            var dn = Generate.Ones(2000);

            var an0 = an.Value[0];
            var bn0 = bn.Value[0];
            if (bn0 != 1)
                throw new Exception("Error in accessor");

            cn.Value[0] = 1;
            dn.Value[0] = 2;

            var cn0 = cn.Value[0];
            var dn0 = dn.Value[0];

            if (cn0 != 1 || dn0 != 2)
                throw new Exception("Error in accessor");

            var test = Generate.Range(3) * 4;
            test.Transpose();

            Shape s = new Shape(
                new long[] { 2, 1, 2, 3 },  //Dimension sizes
                6,                          //Offset
                new long[] { 18, 18, 6, 1 } //Strides
            );

            var a = Generate.Range(s.Length);
            var b = a.Reshape(s);
            var c = b[1][0][1];
            var d = c[2];
            var e = b.Flatten();
            if (e.AsArray().LongLength != 12 || e.AsArray()[3] != 12)
                throw new Exception("Failure in flatten");

            List<T> fln = new List<T>(b[1, 0, 1].Value);
            if (fln.Count != 3) throw new Exception("Failure in basic test 1");
            if (fln[0] != 30) throw new Exception("Failure in basic test 2");
            if (fln[1] != 31) throw new Exception("Failure in basic test 3");
            if (fln[2] != 32) throw new Exception("Failure in basic test 4");

            T n = b.Value[1, 0, 1, 2];
            if (n != 32) throw new Exception("Failure in basic test 5");
            if (c.Value[2] != 32) throw new Exception("Failure in basic test 6");
            if (c.Value[0] != 30) throw new Exception("Failure in basic test 7");
            if (c.Value[1] != 31) throw new Exception("Failure in basic test 8");
            if (b.Sum() != 228) throw new Exception("Failure in basic test 9");

            var r1 = Generate.Range(12).Reshape(new long[] { 2, 1, 2, 3 });
            if (!Equals(r1.AsArray(), new T[] { 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11 })) throw new Exception("Failure in basic test 10");
            var r2 = r1.Reduce<Add>(0);
            if (!Equals(r2.AsArray(), new T[] { 6, 8, 10, 12, 14, 16 })) throw new Exception("Failure in basic test 11");
            r2 = r1.Reduce<Add>(1);
            if (!Equals(r2.AsArray(), new T[] { 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11 })) throw new Exception("Failure in basic test 12");
            r2 = r1.Reduce<Add>(2);
            if (!Equals(r2.AsArray(), new T[] { 3, 5, 7, 15, 17, 19 })) throw new Exception("Failure in basic test 13");
            r2 = r1.Reduce<Add>(3);
            if (!Equals(r2.AsArray(), new T[] { 3, 12, 21, 30 })) throw new Exception(string.Format("Failure in basic test 14"));

            var r3 = b.Reduce<Add>();
            if (!Equals(r3.AsArray(), new T[] { 30, 32, 34, 42, 44, 46 })) throw new Exception("Failure in basic test 15");

            var x1 = Generate.Range(12).Reshape(new long[] { 4, 3 });
            var x2 = Generate.Range(3);

            var x3 = x1 + x2;

            var sqrd = x3.Sqrt();
            sqrd *= sqrd;
            sqrd = sqrd.Round();
            sqrd += 4;
            sqrd++;

            if (UFunc.Reduce<T, Add>(UFunc.Reduce<T, Add>(sqrd, 0)).Value[0] != 138) throw new Exception("Failure in basic arithmetics");
            if (UFunc.Reduce<T, Add>(UFunc.Reduce<T, Add>(sqrd, 1)).Value[0] != 138) throw new Exception("Failure in basic arithmetics");

            var x5 = sqrd.Apply((x) => x % 2 == 0 ? x : -x);
            if (!Equals(x5.AsArray(), new T[] { -5, -7, -9, 8, 10, 12, -11, -13, -15, 14, 16, 18 })) throw new Exception("Failure in basic test 16");

            NumCIL.UFunc.Apply<T, Add>(x1, x2, x3);
            NumCIL.Double.NdArray x4 = (NumCIL.Double.NdArray)x3;
            if (!Equals(x4.AsArray(), new double[] { 0, 2, 4, 3, 5, 7, 6, 8, 10, 9, 11, 13 })) throw new Exception("Failure in basic test 17");

            var x6 = Generate.Range(6).Reshape(new long[] { 2, 3 });

            var x7 = x6.Reduce<Add>();

            var rx7 = x7.AsArray();
            if (!Equals(rx7, new T[] { 3, 5, 7 })) throw new Exception(string.Format("Failure in basic test: [{0}]", string.Join(", ", rx7.Select(x => x.ToString()).ToArray())));

            var x8 = Generate.Range(10) * 0.5f;
            var rx8 = x8.Reduce<Add>().Value[0];
            if (rx8 != 22.5)
                throw new Exception(string.Format("Failure in broadcast multiply: {0}", rx8));

            var x9 = Mul.Apply(Generate.Range(10), 0.5f);
            var rx9 = x9.Reduce<Add>().Value[0];
            if (rx9 != 22.5)
                throw new Exception(string.Format("Failure in broadcast multiply: {0}", rx9));

            var x10 = 5 - Generate.Range(10);
            var x11 = Generate.Range(10) - 5;
            if (x10.Sum() != 5 || x11.Sum() != -5)
                throw new Exception("Failure in scalar rhs/lhs");

            var n0 = Generate.Range(4);
            var n1 = n0[new Range(1, 4)];
            var n2 = n0[new Range(0, 3)];
            var n3 = n1 - n2;
            if (n3.Reduce<Add>().Value[0] != 3)
                throw new Exception("Failure in basic slicing");

            var z0 = Generate.Range(new long[] {2, 2, 3});
            var z1 = z0[Range.All, Range.All, Range.El(1)];
            var z2 = z0[Range.All, Range.El(1), Range.El(1)];
            var z3 = z0[Range.El(1), Range.El(1), Range.All];
            var z4 = z0[Range.El(0), Range.El(1), Range.El(1)];
            var z5 = z0[Range.El(0)];

            if (z1.Shape.Elements != 4)
                throw new Exception("Reduced range failed");
            if (z2.Shape.Elements != 2 || z2.Shape.Dimensions.LongLength != 1)
                throw new Exception("Reduced range failed");
            if (z3.Shape.Elements != 3 || z3.Shape.Dimensions.LongLength != 1)
                throw new Exception("Reduced range failed");
            if (z4.Shape.Elements != 1 || z4.Shape.Dimensions.LongLength != 1)
                throw new Exception("Reduced range failed");
            if (z5.Shape.Elements != 6 || z5.Shape.Dimensions.LongLength != 2)
                throw new Exception("Reduced range failed");

            var y1 = Generate.Range(9).Reshape(new long[] {3, 3});
            var y2 = y1.Transposed;
            y1 = y1.Transposed;
            var y3 = y1 + y2;
            if (y3.Value[0, 2] != 12)
                throw new Exception("Failure with double transpose");

            var y4 = Generate.Range(2 * 2 * 2 * 2 * 2).Reshape(new long[] {2,2,2,2,2});
            var y5 = y4 * (Generate.Range(2) + 1);
            if (y5.Sum() != 752)
                throw new Exception("Failure with 5 dimensions");

            var y6 = Generate.Range(2 * 2 * 2 * 2 * 2 * 2).Reshape(new long[] { 2, 2, 2, 2, 2, 2 });
            var y7 = y6 * (Generate.Range(2) + 1);
            if (y7.Sum() != 3040)
                throw new Exception("Failure with 6 dimensions");

            var r = Generate.Random(2000);
            var rmin = Min.Aggregate(r);
            var rmax = Max.Aggregate(r);
            var ravg = Add.Aggregate(r) / r.DataAccessor.Length;

            if (rmin < 0 || rmax > 1.0)
                throw new Exception("Error in random");
            if (ravg < 0.4 || ravg > 0.6)
                throw new Exception("Deviation in random average");
        }