pymoose.computation.operations module
pymoose.computation.operations module#
- class pymoose.computation.operations.AddNOperation(name, inputs, placement_name, signature)[source]#
- 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]#
- class pymoose.computation.operations.BitwiseOrOperation(name, inputs, placement_name, signature)[source]#
- class pymoose.computation.operations.CastOperation(name, inputs, placement_name, signature)[source]#
- 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]#
- class pymoose.computation.operations.DecryptOperation(name, inputs, placement_name, signature)[source]#
- 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]#
- class pymoose.computation.operations.IdentityOperation(name, inputs, placement_name, signature)[source]#
- 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]#
- class pymoose.computation.operations.InverseOperation(name, inputs, placement_name, signature)[source]#
- class pymoose.computation.operations.LessOperation(name, inputs, placement_name, signature)[source]#
- class pymoose.computation.operations.LoadOperation(name, inputs, placement_name, signature)[source]#
- class pymoose.computation.operations.Log2Operation(name, inputs, placement_name, signature)[source]#
- class pymoose.computation.operations.MaximumOperation(name, inputs, placement_name, signature)[source]#
- 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.OnesOperation(name, inputs, placement_name, signature)[source]#
- 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
- 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]#
- class pymoose.computation.operations.ReshapeOperation(name, inputs, placement_name, signature)[source]#
- class pymoose.computation.operations.SaveOperation(name, inputs, placement_name, signature)[source]#
- 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]#
- class pymoose.computation.operations.SigmoidOperation(name, inputs, placement_name, signature)[source]#
- 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]#
- 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.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]#