shredx.utils.pytorch_polynomial_features#

PyTorch implementation of polynomial feature generation. Drop-in replacement for sklearn.preprocessing.PolynomialFeatures with gradient support.

Classes