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]);
}
}
}
