Skip to content

Commit 7211cf4

Browse files
committed
Test create tensor from generic array.
1 parent 4d4b219 commit 7211cf4

File tree

1 file changed

+7
-3
lines changed

1 file changed

+7
-3
lines changed

Test/TorchSharp.cs

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -78,12 +78,16 @@ public void CreateFloatTensorFromData()
7878
[TestMethod]
7979
public void CreateFloatTensorFromData2()
8080
{
81-
var data = new float[1000];
82-
data[100] = 1;
81+
CreateFloatTensorFromData2Generic<float>();
82+
}
83+
84+
private static void CreateFloatTensorFromData2Generic<T>()
85+
{
86+
var data = new T[1000];
8387

8488
using (var tensor = data.ToTorchTensor(new long[] { 10, 100 }))
8589
{
86-
Assert.AreEqual(tensor.Data[100], 1);
90+
Assert.AreEqual(tensor.Data[100], default(T));
8791
}
8892
}
8993

0 commit comments

Comments
 (0)