pytorch-inference
|
This is the complete list of members for pytorch::Sum, including all inherited members.
dim | pytorch::Sum | private |
forward(const std::vector< tensor > &input) | pytorch::Sum | inlinevirtual |
n_tensors | pytorch::Sum | private |
operator()(const std::vector< tensor > &input) | pytorch::Sum | inlinevirtual |
Sum(const int &dim, const int &n_tensors) | pytorch::Sum | inline |