SR2DynamicMean

Source: models/common/DynamicMean.py

to = mean(from, dim=dynamic_dim). Reduces a named dynamic batch dim.

Mirrors C++ DynamicMean<SR2>. The dim HIT option selects the dynamic-batch axis to average over (the same convention as the C++ Size argument: 0 is the leading dynamic axis, -1 the trailing).

Inputs

from — input · SR2 · required

Input tensor to average

Outputs

to — output · SR2 · required

Averaged tensor

Other options

dim — int · required

The dimension to average over