tltorch.factorized_tensors.init.block_tt_init

tltorch.factorized_tensors.init.block_tt_init(block_tt, std=0.02)[source]

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

Parameters:

block_ttMatrix in the tensor-train format

stdfloat, default is 0.02

the desired standard deviation of the full (reconstructed) tensor

Notes

We assume the given factors form a correct Block-TT decomposition, no checks are done here.