Hello, I wanted to ask a simple question just before I try PyTorch Metal integration.
In tutorial 2 there is a sentence:
"Next we need to make some changes in TorchModule.mm"
Here should be a snippet of code, I guess, that shows how metal() should be used, because next sentence is:
“As you can see, we simply just call .metal() to move our input tensor from CPU to GPU”
So my question is - I should call .metal() on tensor argument just before calling forward method as I call .cuda() for example in Python?