e3j.core.ScalarMixing

class e3j.core.ScalarMixing(source: str | ~e3j.spaces.o3.O3Space, layout: ~e3j.utils.options.Layout = <factory>)

Rescale equivariant features by as many scalars.

This performs the same operation as scalar * irreps_array in e3nn.

Methods

__call__(scalars, features)	Call self as a function.
__init__(source[, layout])

Attributes

num_irreps
source
layout

__call__(scalars: Array, features: Array) → Array

Call self as a function.

__init__(source: str | ~e3j.spaces.o3.O3Space, layout: ~e3j.utils.options.Layout = <factory>) → None