std1.0.0-beta-7 SafeC docs

std :: ml::nn

Function std::tensor_gather_rows

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

&FloatTensor tensor_gather_rows(const &FloatTensor table, const &FloatTensor idx)

out[i] = table[idx[i]]: [K,d] and m indices -> [m,d]. Gradients scatter-add back into the table's rows (embeddings, VQ codebooks).