tltorch.factorized_tensors.init.tensor_init

tltorch.factorized_tensors.init.tensor_init(tensor, std=0.02)[source]

Initializes directly the parameters of a factorized tensor so the reconstruction has the specified standard deviation and 0 mean

Parameters:

tensortorch.Tensor or FactorizedTensor

stdfloat, default is 0.02

the desired standard deviation of the full (reconstructed) tensor