MCPcopy Create free account
hub / github.com/davisking/dlib / softmax_all

Function softmax_all

dlib/cuda/cpu_dlib.cpp:1798–1805  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

1796 // ------------------------------------------------------------------------------------
1797
1798 void softmax_all (
1799 tensor& dest,
1800 const tensor& src
1801 )
1802 {
1803 DLIB_CASSERT(have_same_dimensions(dest,src));
1804 ttimpl::softmax(1, src.nr()*src.nc()*src.k(), dest, src);
1805 }
1806
1807 void softmax_all_gradient (
1808 tensor& grad,

Callers 3

test_softmaxFunction · 0.50
test_softmax_allFunction · 0.50
forward_inplaceMethod · 0.50

Calls 5

have_same_dimensionsFunction · 0.70
softmaxFunction · 0.70
nrMethod · 0.45
ncMethod · 0.45
kMethod · 0.45

Tested by 2

test_softmaxFunction · 0.40
test_softmax_allFunction · 0.40