gorgonia / tensor

package tensor provides efficient and generic n-dimensional arrays in Go that are useful for machine learning and deep learning purposes
Apache License 2.0
359 stars 49 forks source link

Implement Softmax and LogSoftMax #118

Closed dcu closed 2 years ago

dcu commented 2 years ago

backwards operations will be send in a different PR

coveralls commented 2 years ago

Coverage Status

Coverage increased (+0.3%) to 21.55% when pulling f7bcdc48df6aa741d698bfd2280f5b150e57f700 on log-softmax into 1d38d85bc1be32ce4b79e7a2754f39e8d86d9074 on master.

chewxy commented 2 years ago

119 supercedes this