INNER CODE UNIT · Python
default
lucidrains/vector-quantize-pytorch · vector_quantize_pytorch/binary_mapper.py:23
def default(v, d):
return v if exists(v) else d
# tensor helpers
def binary_entropy(logits):
prob = logits.sigmoid()
not_prob = 1. - prob
return -(prob * F.logsigmoid(logits) + not_prob * F.logsigmoid(-logits)).sum(dim = -1)
def pack_with_inverse(t, pattern):
packed, ps = pack([t], pattern)
def inverse(out, inv_pattern = None):
inv_pattern = default(inv_pattern, pattern)
unpacked, = unpack(out, ps, inv_pattern)
return unpacked