Go to the source code of this file.
- Author
- Daniel Nichols
- Version
- 0.1
- Date
- 2019-06-12
- Copyright
- Copyright (c) 2019
◆ crossentropy()
template<typename T >
void magmadnn::math::crossentropy |
( |
Tensor< T > * |
predicted, |
|
|
Tensor< T > * |
ground_truth, |
|
|
Tensor< T > * |
out |
|
) |
| |
Computes the cross entropy of predicted and ground_truth into out.
- Template Parameters
-
- Parameters
-
predicted | |
ground_truth | |
out | scalar tensor |