Engineering Papers⌕ Search

DOE OSTI · code-67479

spdlayers

Abstract

Symmetric Positive Definite (SPD) enforcement layers for PyTorch. Regardless of the input, the output of these layers will always be a SPD tensor! The `Cholesky` layer uses a cholesky factorization to enforce SPD, and the `Eigen` layer uses an eigendecomposition to enforce SPD. Both layers take in some tensor of shape `[batch_size, input_shape]` and output a SPD tensor of shape`[batch_size, output_shape, output_shape]`. The relationship between input and output is defined by the following. ```python input_shape = sum([i for i in range(output_shape + 1)]) ``` The layers have no learnable parameters, and merely serve to transform a vector space to a SPD matrix space.

Keep this discovery

Explore connections, maps & timelines

BibTeXRIS

Jekel, CharlesF.. 2021-09-27. spdlayers. https://doi.org/10.11578/dc.20211129.2

Cite the original work for its findings. Save a collection to share your selection of sources.