jax4j / NablaTensor

Docs / jax4j-core / com.nablatensor.jax4j.nn

final class

MaskedPooling

Presence-aware mean/max pooling over enemy axis 1; defined output for zero enemies. Constants must remain open through the autodiff backward sweep.

View source ↗Full module reference on javadoc.io ↗

Constructors

MaskedPooling(Tensor mask)

Methods

Result forward(Tensor encoded)
DifferentiableResult forward(Var encoded)
void close()