public void CreateIntTensorOnes() { var shape = new long[] { 2, 2 }; TorchTensor t = IntTensor.Ones(shape); Assert.Equal(shape, t.Shape); Assert.Equal(1, t[0, 0].DataItem <int>()); Assert.Equal(1, t[1, 1].DataItem <int>()); }
public void CreateIntTensorOnesCheckData() { var ones = IntTensor.Ones(new long[] { 2, 2 }); var data = ones.Data <int>(); for (int i = 0; i < 4; i++) { Assert.Equal(1, data[i]); } }
public void TestSubInPlace() { var x = IntTensor.Ones(new long[] { 100, 100 }); var y = IntTensor.Ones(new long[] { 100, 100 }); x.SubInPlace(y); var xdata = x.Data <int>(); for (int i = 0; i < 100; i++) { for (int j = 0; j < 100; j++) { Assert.Equal(0, xdata[i + j]); } } }