def call(self, inputs): return sparsemax(inputs, axis=self.axis)
def sparsemax_loss_from_logits(y_true: TensorLike, logits_pred: TensorLike) -> tf.Tensor: y_pred = sparsemax(logits_pred) loss = sparsemax_loss(logits_pred, y_pred, y_true) return loss
def sparsemax_loss_from_logits(y_true, logits_pred): y_pred = sparsemax(logits_pred) loss = sparsemax_loss(logits_pred, y_pred, y_true) return loss