revised stack
This commit is contained in:
+7
-5
@@ -9,14 +9,16 @@ if __name__ == "__main__":
|
||||
|
||||
for n, dim in enumerate(dims):
|
||||
layer = Layer(f"Layer-{n}", dim, params)
|
||||
stack.layer_add(layer)
|
||||
stack.append(layer)
|
||||
|
||||
stack.init(std=0.1)
|
||||
stack.state_save()
|
||||
|
||||
try:
|
||||
stack.layer_add(Layer(f"Layer-{0}", (9,9), params))
|
||||
except StackException as e:
|
||||
print(f"Exception raised: [{e}] -> success!")
|
||||
lay0 = stack.layers[0]
|
||||
lay1 = stack.from_name("Layer-1")
|
||||
lay2 = stack.from_index(2)
|
||||
stack.remove(lay2)
|
||||
stack.remove(lay1)
|
||||
stack.remove(lay0)
|
||||
|
||||
print("Test: [passed]")
|
||||
|
||||
Reference in New Issue
Block a user