MultiplicationCompositionModule

class MultiplicationCompositionModule(*args, **kwargs)[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

func(b)

Multiplies with torch.mul()