Lib.rs

简介

此 crate 提供了对 Tensor 类型的包装器，该类型来自 tensorflow crate，可以创建 ArrayView 和 ArrayViewMut 实例。这使得可以通过 ndarray API 使用张量。

lib.rs:

暴露 ndarray API 的张量包装器。

此 crate 提供了一个小的包装器，用于包装 tensorflow crate 中的 Tensor 数据结构，以便可以使用 ndarray API。此包装器 NdTensor 提供了 view 和 view_mut 方法，分别用于获取 ArrayView 和 ArrayViewMut 实例。

以下示例展示了如何包装一个 Tensor 并获取一个 ArrayView

use ndarray::{arr2, Ix2};
use ndarray_tensorflow::NdTensor;
use tensorflow::Tensor;

let tensor = Tensor::new(&[2, 3])
    .with_values(&[0u32, 1, 2, 3, 4, 5])
    .unwrap();
let array: NdTensor<_, Ix2> =
    NdTensor::from_tensor(tensor)
    .unwrap();
assert_eq!(array.view(),
    arr2(&[[0, 1, 2], [3, 4, 5]]));