File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -12,12 +12,14 @@ namespace stan {
1212namespace math {
1313
1414/* *
15- * Return the log softmax of the specified vector or container of vectors.
15+ * Return the log softmax of the rows of the specified matrix.
16+ * Each row is transformed independently; the result has the same shape
17+ * as the input.
1618 *
17- * @tparam T Type of input vector or matrix.
18- * @param[in] x Unconstrained input vector .
19- * @return Softmax of the input .
20- * @throw std::domain_error If the input vector is size 0.
19+ * @tparam Mat type of input matrix (Eigen matrix with fvar scalar)
20+ * @param[in] m Matrix to transform row-wise .
21+ * @return Log-softmax applied row-wise .
22+ * @throw std::domain_error If the input matrix is size 0.
2123 */
2224template <typename Mat, require_eigen_t <Mat>* = nullptr ,
2325 require_not_eigen_vector_t <Mat>* = nullptr ,
You can’t perform that action at this time.
0 commit comments