pytorch-inference
pytorch::Sum Member List

This is the complete list of members for pytorch::Sum, including all inherited members.

dimpytorch::Sumprivate
forward(const std::vector< tensor > &input)pytorch::Suminlinevirtual
n_tensorspytorch::Sumprivate
operator()(const std::vector< tensor > &input)pytorch::Suminlinevirtual
Sum(const int &dim, const int &n_tensors)pytorch::Suminline