eliott/ort
1
0
Fork 0
mirror of https://github.com/pykeio/ort synced 2026-04-25 16:34:55 +02:00
Code Issues Packages Projects Releases Wiki Activity

feat: add back usize constructors for Shape, ref #370

Browse Source
This commit is contained in:
Carson M.
2025-03-27 10:53:53 -05:00
parent d19e297a9c
commit 317d64bd70
11 changed files with 53 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
examples/custom-ops/custom-ops.rs
View File

@@ -79,14 +79,14 @@ fn main() -> ort::Result<()> {
.commit_from_file("tests/data/custom_op_test.onnx")?;
let allocator = session.allocator();
let mut value1 = Tensor::<f32>::new(allocator, [3, 5])?;
let mut value1 = Tensor::<f32>::new(allocator, [3_usize, 5])?;
{
let (_, data) = value1.extract_tensor_mut();
for datum in data {
*datum = 0.;
}
}
let mut value2 = Tensor::<f32>::new(allocator, [3, 5])?;
let mut value2 = Tensor::<f32>::new(allocator, [3_usize, 5])?;
{
let (_, data) = value2.extract_tensor_mut();
for datum in data {