コード例 #1
0
ファイル: BasicTensorAPI.cs プロジェクト: ganik/TorchSharp
        public void IdentityFloat()
        {
            var x1 = FloatTensor.Eye(10, 10);

            for (int i = 0; i < 10; ++i)
            {
                for (int j = 0; j < 10; ++j)
                {
                    if (i == j)
                    {
                        Assert.AreEqual(1, x1[i, j]);
                    }
                    else
                    {
                        Assert.AreEqual(0, x1[i, j]);
                    }
                }
            }
        }