Operations

Built-in operations

abs = ops.abs[source]
add = ops.add
and_ = ops.and_
eq = ops.eq
exp = ops.exp[source]
ge = ops.ge
getitem = ops.GetitemOp(0)

Op encoding an index into one dimension, e.g. x[:,:,y] for offset of 2.

gt = ops.gt
invert = ops.invert
le = ops.le
log = ops.log[source]
log1p = ops.log1p[source]
lt = ops.lt
matmul = ops.matmul
max = ops.max[source]
min = ops.min[source]
mul = ops.mul
ne = ops.ne
neg = ops.neg
or_ = ops.or_
pow = ops.pow[source]
safediv = ops.safediv[source]
safesub = ops.safesub[source]
sigmoid = ops.sigmoid[source]
sqrt = ops.sqrt[source]
sub = ops.sub
truediv = ops.truediv
xor = ops.xor

Operation classes

class Op(fn)[source]

Bases: multipledispatch.dispatcher.Dispatcher

class AssociativeOp(fn)[source]

Bases: funsor.ops.Op

class AddOp(fn)[source]

Bases: funsor.ops.AssociativeOp

class LogAddExpOp(fn)[source]

Bases: funsor.ops.AssociativeOp

class SubOp(fn)[source]

Bases: funsor.ops.Op

class NegOp(fn)[source]

Bases: funsor.ops.Op

class ReshapeOp(shape)[source]

Bases: funsor.ops.Op

class GetitemOp(offset)[source]

Bases: funsor.ops.Op

Op encoding an index into one dimension, e.g. x[:,:,y] for offset of 2.

class ExpOp(fn)[source]

Bases: funsor.ops.TransformOp

class LogOp(fn)[source]

Bases: funsor.ops.TransformOp

class ReciprocalOp(fn)[source]

Bases: funsor.ops.Op