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

Get shape to return a tuple.

Browse Source
This commit is contained in:
laurent
2023-07-01 21:34:38 +01:00
parent fbbde5b02c
commit 86df4ad79c
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
candle-pyo3/src/lib.rs
View File

@ -1,3 +1,4 @@
use pyo3::types::PyTuple;
use pyo3::exceptions::{PyTypeError, PyValueError};
use pyo3::prelude::*;
@ -27,8 +28,8 @@ impl PyTensor {
}
#[getter]
fn shape(&self) -> Vec<usize> {
self.0.dims().to_vec()
fn shape(&self, py: Python<'_>) -> PyObject {
PyTuple::new(py, self.0.dims()).to_object(py)
}
#[getter]