MatrixMultiplier Struct Reference

Public Member Functions
	MatrixMultiplier (int A, int B)
torch::Tensor	forward (torch::Tensor weights)
torch::Tensor	get () const

Detailed Description

Definition at line 7 of file extension.cpp.

---

The documentation for this struct was generated from the following file:

• test/cpp_extensions/extension.cpp