pyvene.models.backpack_gpt2.modelings_backpack_gpt2.BackpackGPT2LMHeadModelOutput

class BackpackGPT2LMHeadModelOutput(logits: torch.FloatTensor = None, contextualization: torch.FloatTensor = None)

Bases: ModelOutput

__init__(logits: FloatTensor | None = None, contextualization: FloatTensor | None = None) → None

Methods

__init__([logits, contextualization])
clear()
copy()
fromkeys([value])	Create a new ordered dictionary with keys from iterable and values set to value.
get(key[, default])	Return the value for key if key is in the dictionary, else default.
items()
keys()
move_to_end(key[, last])	Move an existing element to the end (or beginning if last is false).
pop(key[,default])	If the key is not found, return the default if given; otherwise, raise a KeyError.
popitem([last])	Remove and return a (key, value) pair from the dictionary.
setdefault(*args, **kwargs)	Insert key with a value of default if key is not in the dictionary.
to_tuple()	Convert self to a tuple containing all the attributes/keys that are not None.
update([E, ]**F)	If E is present and has a .keys() method, then does: for k in E: D[k] = E[k] If E is present and lacks a .keys() method, then does: for k, v in E: D[k] = v In either case, this is followed by: for k in F: D[k] = F[k]
values()

Attributes

contextualization
logits