Module adseq.implementations.ring

Classes

class Ring (buffer: jax.Array)

Expand source code

class Ring(typing.NamedTuple):
    buffer: jax.Array
    @classmethod
    def init(cls, delay, grad=False):
        return cls(jnp.full(
            delay+1,
            0,
            'float32' if grad else 'int32'))
    def enqueue(self, n):
        return _enqueue(self, n)
    def pop(self, n):
        return _pop(self, n)

Ring(buffer,)

Ancestors

• builtins.tuple

Static methods

def init(delay, grad=False)

Instance variables

var buffer : jax.Array

Expand source code

class Ring(typing.NamedTuple):
    buffer: jax.Array
    @classmethod
    def init(cls, delay, grad=False):
        return cls(jnp.full(
            delay+1,
            0,
            'float32' if grad else 'int32'))
    def enqueue(self, n):
        return _enqueue(self, n)
    def pop(self, n):
        return _pop(self, n)

Alias for field number 0

Methods

def enqueue(self, n)

Expand source code

def enqueue(self, n):
    return _enqueue(self, n)

def pop(self, n)

Expand source code

def pop(self, n):
    return _pop(self, n)