We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent f8171e1 commit 3169921Copy full SHA for 3169921
1 file changed
rustorch-pytorch/src/lib.rs
@@ -122,7 +122,9 @@ mod tests {
122
// Convert to PyTorch
123
let torch_tensor = PyTorchAdapter::to_torch(&rt_tensor);
124
assert_eq!(torch_tensor.size(), vec![2, 2]);
125
- let data_vec: Vec<f32> = torch_tensor.shallow_clone().into();
+ let numel = torch_tensor.numel();
126
+ let mut data_vec = vec![0.0f32; numel as usize];
127
+ torch_tensor.copy_data(&mut data_vec, numel as usize);
128
assert_eq!(data_vec, data);
129
130
// Convert back
0 commit comments