Expose candle gather op in pyo3. (#1870)

This commit is contained in:
Laurent Mazare
2024-03-18 21:54:15 +01:00
committed by GitHub
parent f115895b9e
commit 143c481c20
2 changed files with 12 additions and 0 deletions

View File

@ -324,6 +324,12 @@ class Tensor:
"""
pass
def gather(self, index, dim):
"""
Gathers values along an axis specified by dim.
"""
pass
def get(self, index: int) -> Tensor:
"""
Gets the value at the specified index.