pyjuice.set_block_size

class pyjuice.set_block_size(block_size: int = 1)

Context-manager that sets the block size of PC nodes constructed by inputs, multiply, and summate.

Parameters:

block_size (int) – the target block size

Example::

>>> with pyjuice.set_block_size(16):
...     nis = pyjuice.inputs(var = 0, num_node_blocks = 4, dist = Categorical(num_cats = 20))
>>> print(nis.num_nodes) # Should be 16 * 4
64

__init__(block_size: int = 1)