peal.teachers.spray_teacher.LogitDifference

class peal.teachers.spray_teacher.LogitDifference[source]

Bases: Module

Module returning y[0] - y[1]; a binary logit-difference head.

Methods

forward

Return the difference of the first two entries of y.

__init__()[source]

Stateless module; only calls nn.Module.__init__.

forward(y)[source]

Return the difference of the first two entries of y.