pytorch-inference
pytorch::Difference Member List

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

Difference(const int &dim, const int &n_tensors)pytorch::Differenceinline
dimpytorch::Differenceprivate
forward(const std::vector< tensor > &input)pytorch::Differenceinlinevirtual
n_tensorspytorch::Differenceprivate
operator()(const std::vector< tensor > &input)pytorch::Differenceinlinevirtual