bte.utils.trace#

Module Contents#

Functions#

trace2d(T)

tensor2d_tosymmetry(T)

tensor2d_totrace0(T[, need_symmetrize])

tensor3d_tosymmetry(T)

tensor3d_totrace0(T[, need_symmetrize])

bte.utils.trace.trace2d(T)[source]#
Parameters:

T (torch.Tensor) –

bte.utils.trace.tensor2d_tosymmetry(T)[source]#
Parameters:

T (torch.Tensor) –

bte.utils.trace.tensor2d_totrace0(T, need_symmetrize=True)[source]#
Parameters:

T (torch.Tensor) –

bte.utils.trace.tensor3d_tosymmetry(T)[source]#
Parameters:

T (torch.Tensor) –

bte.utils.trace.tensor3d_totrace0(T, need_symmetrize=True)[source]#
Parameters:

T (torch.Tensor) –