@@ -120,18 +120,6 @@ final class TensorTests: XCTestCase {
120120 )
121121 }
122122
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-
135123 func testAnnotationsTFEager( ) {
136124 let tensor = Tensor < Float > ( repeating: 0 , shape: [ 1 , 2 , 3 ] , on: Device . defaultTFEager)
137125 XCTAssertEqual ( tensor. annotations, " Annotations not available in TF_EAGER. " )
@@ -148,7 +136,6 @@ final class TensorTests: XCTestCase {
148136 ( " testTensorShapeCollectionOperations " , testTensorShapeCollectionOperations) ,
149137 ( " testInitShapeScalars " , testInitShapeScalars) ,
150138 ( " testInitShapeScalarsDerivative " , testInitShapeScalarsDerivative) ,
151- ( " testZeroTangentVectorInitializer " , testZeroTangentVectorInitializer) ,
152139 ( " testAnnotationsTFEager " , testAnnotationsTFEager) ,
153140 ]
154141}
0 commit comments