tltorch.init.cp_init

tltorch.init.cp_init(weights, factors, std=0.02)[source]

Initializes directly the weights and factors of a CP decomposition so the reconstruction has the specified std and 0 mean

Parameters:
weights1D tensor
factorslist of 2D factors of size (dim_i, rank)
stdfloat, default is 0.02

the desired standard deviation of the full (reconstructed) tensor

Notes

We assume the given (weights, factors) form a correct CP decomposition, no checks are done here.