@@ -92,7 +92,7 @@ public extension Conv1D where Scalar.RawSignificand: FixedWidthInteger {
92
92
filterShape. 0 , filterShape. 1 , filterShape. 2 ] )
93
93
self . init (
94
94
filter: Tensor ( glorotUniform: filterTensorShape) ,
95
- bias: Tensor ( zeros: TensorShape ( [ filterShape. 2 ] ) ) ,
95
+ bias: Tensor ( zeros: [ filterShape. 2 ] ) ,
96
96
activation: activation,
97
97
stride: stride,
98
98
padding: padding)
@@ -123,7 +123,7 @@ public extension Conv1D {
123
123
filterShape. 0 , filterShape. 1 , filterShape. 2 ] )
124
124
self . init (
125
125
filter: Tensor ( glorotUniform: filterTensorShape, seed: seed) ,
126
- bias: Tensor ( zeros: TensorShape ( [ filterShape. 2 ] ) ) ,
126
+ bias: Tensor ( zeros: [ filterShape. 2 ] ) ,
127
127
activation: activation,
128
128
stride: stride,
129
129
padding: padding)
@@ -208,7 +208,7 @@ public extension Conv2D {
208
208
filterShape. 0 , filterShape. 1 , filterShape. 2 , filterShape. 3 ] )
209
209
self . init (
210
210
filter: Tensor ( glorotUniform: filterTensorShape, generator: & generator) ,
211
- bias: Tensor ( zeros: TensorShape ( [ filterShape. 3 ] ) ) ,
211
+ bias: Tensor ( zeros: [ filterShape. 3 ] ) ,
212
212
activation: activation,
213
213
strides: strides,
214
214
padding: padding)
@@ -238,7 +238,7 @@ public extension Conv2D {
238
238
filterShape. 0 , filterShape. 1 , filterShape. 2 , filterShape. 3 ] )
239
239
self . init (
240
240
filter: Tensor ( glorotUniform: filterTensorShape, seed: seed) ,
241
- bias: Tensor ( zeros: TensorShape ( [ filterShape. 3 ] ) ) ,
241
+ bias: Tensor ( zeros: [ filterShape. 3 ] ) ,
242
242
activation: activation,
243
243
strides: strides,
244
244
padding: padding)
@@ -324,7 +324,7 @@ public extension Conv3D {
324
324
filterShape. 0 , filterShape. 1 , filterShape. 2 , filterShape. 3 , filterShape. 4 ] )
325
325
self . init (
326
326
filter: Tensor ( glorotUniform: filterTensorShape, generator: & generator) ,
327
- bias: Tensor ( zeros: TensorShape ( [ filterShape. 4 ] ) ) ,
327
+ bias: Tensor ( zeros: [ filterShape. 4 ] ) ,
328
328
activation: activation,
329
329
strides: strides,
330
330
padding: padding)
@@ -354,7 +354,7 @@ public extension Conv3D {
354
354
filterShape. 0 , filterShape. 1 , filterShape. 2 , filterShape. 3 , filterShape. 4 ] )
355
355
self . init (
356
356
filter: Tensor ( glorotUniform: filterTensorShape, seed: seed) ,
357
- bias: Tensor ( zeros: TensorShape ( [ filterShape. 4 ] ) ) ,
357
+ bias: Tensor ( zeros: [ filterShape. 4 ] ) ,
358
358
activation: activation,
359
359
strides: strides,
360
360
padding: padding)
@@ -449,7 +449,7 @@ public extension TransposedConv2D {
449
449
filterShape. 0 , filterShape. 1 , filterShape. 2 , filterShape. 3 ] )
450
450
self . init (
451
451
filter: Tensor ( glorotUniform: filterTensorShape, generator: & generator) ,
452
- bias: Tensor ( zeros: TensorShape ( [ filterShape. 3 ] ) ) ,
452
+ bias: Tensor ( zeros: [ filterShape. 3 ] ) ,
453
453
activation: activation,
454
454
strides: strides,
455
455
padding: padding)
@@ -479,7 +479,7 @@ public extension TransposedConv2D {
479
479
filterShape. 0 , filterShape. 1 , filterShape. 2 , filterShape. 3 ] )
480
480
self . init (
481
481
filter: Tensor ( glorotUniform: filterTensorShape, seed: seed) ,
482
- bias: Tensor ( zeros: TensorShape ( [ filterShape. 3 ] ) ) ,
482
+ bias: Tensor ( zeros: [ filterShape. 3 ] ) ,
483
483
activation: activation,
484
484
strides: strides,
485
485
padding: padding)
0 commit comments