Function build_info
pub fn build_info<R, E>(tensors: &[&JitTensor<R>]) -> Vec<u32>where
R: JitRuntime,
E: JitElement,Expand description
Create a vector containing the dimension, strides and shape of tensors.
ยงExample
With two tensors (lhs, rhs)
| Indexes | Value |
|---|---|
| 0..1 | D |
| 1..D + 1 | lhs strides |
| (D + 1)..(2 * D + 1) | rhs strides |
| (2 * D + 1)..(3 * D + 1) | lhs shape |
| (3 * D + 1)..(4 * D + 1) | rhs shape |