Trait coaster_nn::LogSoftmax
[−]
[src]
pub trait LogSoftmax<F>: NN<F> { fn log_softmax(
&self,
x: &SharedTensor<F>,
result: &mut SharedTensor<F>
) -> Result<(), Error>; fn log_softmax_grad(
&self,
x: &SharedTensor<F>,
x_diff: &SharedTensor<F>,
result_diff: &mut SharedTensor<F>
) -> Result<(), Error>; }
Provides the functionality for a Backend to support LogSoftmax operations.
Required Methods
fn log_softmax(
&self,
x: &SharedTensor<F>,
result: &mut SharedTensor<F>
) -> Result<(), Error>
&self,
x: &SharedTensor<F>,
result: &mut SharedTensor<F>
) -> Result<(), Error>
Computes a logarithmic softmax over the input Tensor x
.
Saves the result to result
.
fn log_softmax_grad(
&self,
x: &SharedTensor<F>,
x_diff: &SharedTensor<F>,
result_diff: &mut SharedTensor<F>
) -> Result<(), Error>
&self,
x: &SharedTensor<F>,
x_diff: &SharedTensor<F>,
result_diff: &mut SharedTensor<F>
) -> Result<(), Error>
Computes the gradient of a logarithmic softmax over the input Tensor x
.
Saves the result to result_diff
.