beatmup.nnets.Softmax

class beatmup.nnets.Softmax

Softmax layer. It does not have output, but acts as a sink. The resulting probabilities are returned by getProbabilities(). This operation is executed on CPU.

__init__(self: beatmup.nnets.Softmax, name: str) → None

Creates a softmax layer.

Parameters:

name – operation name

Methods

__init__(self, name)

Creates a softmax layer.

get_probabilities(self)

Returns the list of probabilities

Attributes

input_count

Number of operation inputs

name

Operation name

output_count

Number of operation outputs