Complete tensor ops #16
Labels
enhancement
New feature or request
good first issue
Good for newcomers
help wanted
Extra attention is needed
We should implement all ops available as methods on a tensor that make sense in Tensor.
Right now, quite a few are still missing, and even for the ones already ported, many still need docs.
Many of the calls are mostly wrappers of the native call and return a tensor of the same type, so they should be straightforward to add if you look at the existing ones, i.e.
unsqueeze
.storch/core/src/main/scala/torch/Tensor.scala
Lines 428 to 432 in 2ea690f
Some need a bit more effort, i.e. if they return a different tensor type.
For Scaladoc it can be useful to look at the original PyTorch doc sources for these ops. Note that these docs are for the
torch.unsqueeze(tensor, ...)
but I think we should focus on the method defined onTensor
itself likemyTensor.unsqueeze(...)
first. We can still add the function variant later.The text was updated successfully, but these errors were encountered: