11 ms·

If you want to do this today you can also use the torch c++ api! It’s whats pytorch binds to under the hood.

• whimsicalism 3y ago
  ? I don't think torch C++ supports this.