Integrating Tensor Initialization with Custom Functions Post-Decomposition

I’m exploring ways to incorporate tensor initialization using T.init() into my custom functions after executing decompose_reduction. I’ve been searching for examples but haven’t found any that match my needs. Specifically, I’m looking to initialize computation results with a bias. Has anyone here successfully implemented this, or can point me towards some resources or examples? Any guidance would be greatly appreciated!