pymoose.computation.operations module#

class pymoose.computation.operations.AbsOperation(name, inputs, placement_name, signature)[source]#

Bases: pymoose.computation.operations.Operation

class pymoose.computation.operations.AddNOperation(name, inputs, placement_name, signature)[source]#

Bases: pymoose.computation.operations.Operation

class pymoose.computation.operations.AddOperation(name, inputs, placement_name, signature)[source]#

Bases: pymoose.computation.operations.Operation

class pymoose.computation.operations.ArgmaxOperation(name, inputs, placement_name, signature, axis, upmost_index)[source]#

Bases: pymoose.computation.operations.Operation

axis: Optional[Tuple[int]]#
upmost_index: int#
class pymoose.computation.operations.AtLeast2DOperation(name, inputs, placement_name, signature, to_column_vector)[source]#

Bases: pymoose.computation.operations.Operation

to_column_vector: bool#
class pymoose.computation.operations.BitwiseAndOperation(name, inputs, placement_name, signature)[source]#

Bases: pymoose.computation.operations.Operation

class pymoose.computation.operations.BitwiseOrOperation(name, inputs, placement_name, signature)[source]#

Bases: pymoose.computation.operations.Operation

class pymoose.computation.operations.CastOperation(name, inputs, placement_name, signature)[source]#

Bases: pymoose.computation.operations.Operation

class pymoose.computation.operations.ConcatenateOperation(name, inputs, placement_name, signature, axis)[source]#

Bases: pymoose.computation.operations.Operation

axis: int#
class pymoose.computation.operations.ConstantOperation(name, inputs, placement_name, signature, value)[source]#

Bases: pymoose.computation.operations.Operation

value: pymoose.computation.values.Value#
class pymoose.computation.operations.DecryptOperation(name, inputs, placement_name, signature)[source]#

Bases: pymoose.computation.operations.Operation

class pymoose.computation.operations.DivOperation(name, inputs, placement_name, signature)[source]#

Bases: pymoose.computation.operations.Operation

class pymoose.computation.operations.DotOperation(name, inputs, placement_name, signature)[source]#

Bases: pymoose.computation.operations.Operation

class pymoose.computation.operations.ExpOperation(name, inputs, placement_name, signature)[source]#

Bases: pymoose.computation.operations.Operation

class pymoose.computation.operations.ExpandDimsOperation(name, inputs, placement_name, signature, axis)[source]#

Bases: pymoose.computation.operations.Operation

axis: Tuple[int]#
class pymoose.computation.operations.GreaterOperation(name, inputs, placement_name, signature)[source]#

Bases: pymoose.computation.operations.Operation

class pymoose.computation.operations.IdentityOperation(name, inputs, placement_name, signature)[source]#

Bases: pymoose.computation.operations.Operation

class pymoose.computation.operations.IndexAxisOperation(name, inputs, placement_name, signature, axis, index)[source]#

Bases: pymoose.computation.operations.Operation

axis: int#
index: int#
class pymoose.computation.operations.InputOperation(name, inputs, placement_name, signature)[source]#

Bases: pymoose.computation.operations.Operation

class pymoose.computation.operations.InverseOperation(name, inputs, placement_name, signature)[source]#

Bases: pymoose.computation.operations.Operation

class pymoose.computation.operations.LessOperation(name, inputs, placement_name, signature)[source]#

Bases: pymoose.computation.operations.Operation

class pymoose.computation.operations.LoadOperation(name, inputs, placement_name, signature)[source]#

Bases: pymoose.computation.operations.Operation

class pymoose.computation.operations.Log2Operation(name, inputs, placement_name, signature)[source]#

Bases: pymoose.computation.operations.Operation

class pymoose.computation.operations.LogOperation(name, inputs, placement_name, signature)[source]#

Bases: pymoose.computation.operations.Operation

class pymoose.computation.operations.MaximumOperation(name, inputs, placement_name, signature)[source]#

Bases: pymoose.computation.operations.Operation

class pymoose.computation.operations.MeanOperation(name, inputs, placement_name, signature, axis)[source]#

Bases: pymoose.computation.operations.Operation

axis: Optional[Union[int, Tuple[int]]]#
class pymoose.computation.operations.MulOperation(name, inputs, placement_name, signature)[source]#

Bases: pymoose.computation.operations.Operation

class pymoose.computation.operations.MuxOperation(name, inputs, placement_name, signature)[source]#

Bases: pymoose.computation.operations.Operation

class pymoose.computation.operations.OnesOperation(name, inputs, placement_name, signature)[source]#

Bases: pymoose.computation.operations.Operation

class pymoose.computation.operations.OpSignature(input_types, return_type)[source]#

Bases: object

input_types: Dict[str, pymoose.computation.types.ValueType]#
return_type: pymoose.computation.types.ValueType#
class pymoose.computation.operations.Operation[source]#

Bases: object

classmethod identifier()[source]#
inputs: Dict[str, str]#
name: str#
placement_name: str#
property return_type#
signature: pymoose.computation.operations.OpSignature#
class pymoose.computation.operations.OutputOperation(name, inputs, placement_name, signature, tag)[source]#

Bases: pymoose.computation.operations.Operation

tag: str#
class pymoose.computation.operations.ReluOperation(name, inputs, placement_name, signature)[source]#

Bases: pymoose.computation.operations.Operation

class pymoose.computation.operations.ReshapeOperation(name, inputs, placement_name, signature)[source]#

Bases: pymoose.computation.operations.Operation

class pymoose.computation.operations.SaveOperation(name, inputs, placement_name, signature)[source]#

Bases: pymoose.computation.operations.Operation

class pymoose.computation.operations.SelectOperation(name, inputs, placement_name, signature, axis)[source]#

Bases: pymoose.computation.operations.Operation

axis: int#
class pymoose.computation.operations.ShapeOperation(name, inputs, placement_name, signature)[source]#

Bases: pymoose.computation.operations.Operation

class pymoose.computation.operations.SigmoidOperation(name, inputs, placement_name, signature)[source]#

Bases: pymoose.computation.operations.Operation

class pymoose.computation.operations.SliceOperation(name, inputs, placement_name, signature, begin, end)[source]#

Bases: pymoose.computation.operations.Operation

begin: int#
end: int#
class pymoose.computation.operations.SoftmaxOperation(name, inputs, placement_name, signature, axis, upmost_index)[source]#

Bases: pymoose.computation.operations.Operation

axis: Optional[Tuple[int]]#
upmost_index: int#
class pymoose.computation.operations.SqrtOperation(name, inputs, placement_name, signature)[source]#

Bases: pymoose.computation.operations.Operation

class pymoose.computation.operations.SqueezeOperation(name, inputs, placement_name, signature, axis)[source]#

Bases: pymoose.computation.operations.Operation

axis: Optional[Union[int, Tuple[int]]]#
class pymoose.computation.operations.StridedSliceOperation(name, inputs, placement_name, signature, slices)[source]#

Bases: pymoose.computation.operations.Operation

slices: Optional[Tuple[slice]]#
class pymoose.computation.operations.SubOperation(name, inputs, placement_name, signature)[source]#

Bases: pymoose.computation.operations.Operation

class pymoose.computation.operations.SumOperation(name, inputs, placement_name, signature, axis)[source]#

Bases: pymoose.computation.operations.Operation

axis: Optional[Union[int, Tuple[int]]]#
class pymoose.computation.operations.TransposeOperation(name, inputs, placement_name, signature)[source]#

Bases: pymoose.computation.operations.Operation

class pymoose.computation.operations.ZerosOperation(name, inputs, placement_name, signature)[source]#

Bases: pymoose.computation.operations.Operation