`elasticai.creator.nn.binary.math_operations`#

Module Contents#

Classes#

MathOperations

API#

class elasticai.creator.nn.binary.math_operations.MathOperations[source]#

Bases: elasticai.creator.base_modules.linear.MathOperations, elasticai.creator.base_modules.conv1d.MathOperations, elasticai.creator.base_modules.lstm_cell.MathOperations

quantize(a: torch.Tensor) → torch.Tensor[source]#

add(a: torch.Tensor, b: torch.Tensor) → torch.Tensor[source]#

mul(a: torch.Tensor, b: torch.Tensor) → torch.Tensor[source]#

matmul(a: torch.Tensor, b: torch.Tensor) → torch.Tensor[source]#