MultiplicationCompositionModule

class MultiplicationCompositionModule(*args: Any, **kwargs: Any)[source]

Bases: FunctionalCompositionModule

Composition by element-wise multiplication.

Initialize internal Module state, shared by both nn.Module and ScriptModule.

Methods Summary

func(b)

Multiplies with torch.mul()

Methods Documentation

Parameters:
func(b)

Multiplies with torch.mul()