huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-18 19:47:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

convert pytorch's tensor in Python API (#1172)

Browse Source 
* convert pytorch's tensor

* separate tests for convert pytorch tensor
This commit is contained in:
andrew
2023-10-26 01:39:14 +07:00
committed by GitHub
parent 0acd16751d
commit 6a446d9d73
3 changed files with 43 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
candle-pyo3/py_src/candle/__init__.pyi
View File

@ -396,6 +396,11 @@ class Tensor:
Convert the tensor to a new dtype.
"""
pass
def to_torch(self) -> torch.Tensor:
"""
Converts candle's tensor to pytorch's tensor
"""
pass
def transpose(self, dim1: int, dim2: int) -> Tensor:
"""
Returns a tensor that is a transposed version of the input, the given dimensions are swapped.