grain.functions

Members

Structs

AddBias
struct AddBias(T)

TODO generalize to broadcastable addition

LogSoftmax
struct LogSoftmax(T, size_t dim = 2)
MatMul
struct MatMul(T)

Matrix-Matrix multiplication

NegativeLogLikelihood
struct NegativeLogLikelihood(F, I = long)

Meta