@@ -15,19 +15,6 @@ public void CreateFloatTensorOnes()
15
15
Assert . IsNotNull ( ones ) ;
16
16
}
17
17
18
- [ TestMethod ]
19
- public void CreateFloatTensorCheckDistructor ( )
20
- {
21
- ITorchTensor < float > ones = null ;
22
-
23
- using ( var tmp = FloatTensor . Ones ( new long [ ] { 2 , 2 } ) )
24
- {
25
- ones = tmp ;
26
- Assert . IsNotNull ( ones ) ;
27
- }
28
- Assert . ThrowsException < ObjectDisposedException > ( ones . Grad ) ;
29
- }
30
-
31
18
[ TestMethod ]
32
19
public void CreateFloatTensorCheckMemory ( )
33
20
{
@@ -76,6 +63,18 @@ public void CreateFloatTensorCheckDevice()
76
63
Assert . AreEqual ( ones . Device , "cpu" ) ;
77
64
}
78
65
66
+ [ TestMethod ]
67
+ public void CreateFloatTensorFromData ( )
68
+ {
69
+ var data = new float [ 1000 ] ;
70
+ data [ 100 ] = 1 ;
71
+
72
+ using ( var tensor = FloatTensor . From ( data , new long [ ] { 100 , 10 } , new long [ ] { 1 , 100 } ) )
73
+ {
74
+ Assert . AreEqual ( tensor . Data [ 100 ] , 1 ) ;
75
+ }
76
+ }
77
+
79
78
[ TestMethod ]
80
79
public void ScoreModel ( )
81
80
{
@@ -90,7 +89,7 @@ public void ScoreModel()
90
89
}
91
90
92
91
[ TestMethod ]
93
- public void ScoreModelCheckInput ( )
92
+ public void LoadModelCheckInput ( )
94
93
{
95
94
var module = JIT . Module . Load ( @"E:\Source\Repos\libtorch\model.pt" ) ;
96
95
Assert . IsNotNull ( module ) ;
@@ -105,6 +104,22 @@ public void ScoreModelCheckInput()
105
104
}
106
105
}
107
106
107
+ [ TestMethod ]
108
+ public void LoadModelCheckOutput ( )
109
+ {
110
+ var module = JIT . Module . Load ( @"E:\Source\Repos\libtorch\model.pt" ) ;
111
+ Assert . IsNotNull ( module ) ;
112
+
113
+ var num = module . GetNumberOfOutputs ( ) ;
114
+
115
+ for ( int i = 0 ; i < num ; i ++ )
116
+ {
117
+ var type = module . GetOutputType ( i ) ;
118
+
119
+ Assert . IsNotNull ( type as DynamicType ) ;
120
+ }
121
+ }
122
+
108
123
[ TestMethod ]
109
124
public void ScoreModelCheckOutput ( )
110
125
{
0 commit comments