Skip to content

Commit 974855c

Browse files
committed
fix BasicConv1D unit test
1 parent cb4e4b0 commit 974855c

File tree

2 files changed

+3
-5
lines changed

2 files changed

+3
-5
lines changed

src/TensorFlowNET.Keras/Layers/LayersApi.cs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -85,7 +85,7 @@ public BatchNormalization BatchNormalization(int axis = -1,
8585
/// <param name="bias_initializer">Initializer for the bias vector (see keras.initializers).</param>
8686
/// <returns>A tensor of rank 3 representing activation(conv1d(inputs, kernel) + bias).</returns>
8787
public Conv1D Conv1D(int filters,
88-
Shape? kernel_size = null,
88+
Shape kernel_size,
8989
int? strides = null,
9090
string padding = "valid",
9191
string data_format = null,

test/TensorFlowNET.Keras.UnitTest/Layers/Layers.Convolution.Test.cs

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -14,14 +14,12 @@ public void BasicConv1D()
1414
{
1515
var filters = 8;
1616

17-
var conv = keras.layers.Conv1D(filters, activation: "linear");
17+
var conv = keras.layers.Conv1D(filters, kernel_size: 3, activation: "linear");
1818

1919
var x = np.arange(256.0f).reshape((8, 8, 4));
2020
var y = conv.Apply(x);
2121

22-
Assert.AreEqual(3, y.shape.ndim);
23-
Assert.AreEqual(x.dims[0], y.shape[0]);
24-
Assert.AreEqual(x.dims[1] - 4, y.shape[1]);
22+
Assert.AreEqual(y.shape, (8, 6, 8));
2523
Assert.AreEqual(filters, y.shape[2]);
2624
}
2725

0 commit comments

Comments
 (0)