std1.0.0-beta-7 SafeC docs

std :: ml::sparse

Function std::tensor_spmm

#include <std/ml/sparse.h> source ยท definition

&FloatTensor tensor_spmm(const &SparseTensor s, const &FloatTensor x)

out [rows, n] = S [rows, cols] . X [cols, n], with gradients to S's values and to X. Runs on the device of S's values / X when either is on one.