Tannic
A C++ Tensor Library
|
This is the complete list of members for tannic::expression::Argmax, including all inherited members.
axis | tannic::expression::Argmax | |
forward(Tensor const &input, Tensor &output) const | tannic::expression::Argmax | |
keepdim | tannic::expression::Argmax | |
reduce(type dtype) const | tannic::expression::Argmax | inline |
reduce(Shape const &shape) const | tannic::expression::Argmax | inline |