@@ -120,18 +120,6 @@ final class TensorTests: XCTestCase {
120
120
)
121
121
}
122
122
123
- func testZeroTangentVectorInitializer( ) {
124
- let shape : TensorShape = [ 4 , 5 , 6 ]
125
- let tensor = Tensor < Float > ( randomUniform: shape)
126
- XCTAssertEqual ( tensor. zeroTangentVector, Tensor ( zeros: shape) )
127
-
128
- struct TensorWrapper : Differentiable {
129
- var tensor : Tensor < Float >
130
- }
131
- let model = TensorWrapper ( tensor: tensor)
132
- XCTAssertEqual ( model. zeroTangentVector, . init( tensor: Tensor ( zeros: shape) ) )
133
- }
134
-
135
123
func testAnnotationsTFEager( ) {
136
124
let tensor = Tensor < Float > ( repeating: 0 , shape: [ 1 , 2 , 3 ] , on: Device . defaultTFEager)
137
125
XCTAssertEqual ( tensor. annotations, " Annotations not available in TF_EAGER. " )
@@ -148,7 +136,6 @@ final class TensorTests: XCTestCase {
148
136
( " testTensorShapeCollectionOperations " , testTensorShapeCollectionOperations) ,
149
137
( " testInitShapeScalars " , testInitShapeScalars) ,
150
138
( " testInitShapeScalarsDerivative " , testInitShapeScalarsDerivative) ,
151
- ( " testZeroTangentVectorInitializer " , testZeroTangentVectorInitializer) ,
152
139
( " testAnnotationsTFEager " , testAnnotationsTFEager) ,
153
140
]
154
141
}
0 commit comments