コード例 #1
0
ファイル: UnaryTensorOps.cs プロジェクト: ganik/TorchSharp
        public void CreateFloatTensorLike()
        {
            var x1 = new FloatTensor(200, 200);
            var x2 = x1.OnesLike();
            var x3 = x1.ZerosLike();

            Assert.IsNotNull(x1);
            Assert.IsNotNull(x2);
            Assert.IsNotNull(x3);
            Assert.AreEqual(x1.Shape.Length, x2.Shape.Length);
            Assert.AreEqual(x1.Shape.Length, x3.Shape.Length);

            x2.Resize1d(200 * 200);
            x3.Resize1d(200 * 200);
            for (var i = 0; i < x2.Shape[0]; ++i)
            {
                Assert.AreEqual(1, x2[i]);
                Assert.AreEqual(0, x3[i]);
            }
        }