jens/pyRBM
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

added shape to entity

Browse Source
This commit is contained in:
jens
2025-12-18 19:47:27 +01:00
parent c1c6c610ad
commit 6664ebdb9b
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/rbm/entity.py
+1
View File
@@ -6,6 +6,7 @@ from status import Status
class Entity: class Entity:
def __init__(self, shape: tuple[int, int], params: RbmParams): def __init__(self, shape: tuple[int, int], params: RbmParams):
self.shape = shape
self.state = RbmState.from_layer_params(shape) self.state = RbmState.from_layer_params(shape)
self.params = params self.params = params