snippets/snippets/
tensor_simple.rs

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
//! DO NOT EDIT! This file was autogenerated by `docs/snippets/build.rs`. The original is in `/home/runner/work/rerun/rerun/docs/snippets/all/archetypes/tensor_simple.rs`.
//! Create and log a tensor.

use ndarray::{Array, ShapeBuilder as _};
use rand::prelude::*;

pub fn main(_args: &[String]) -> Result<(), Box<dyn std::error::Error>> {
    let rec = rerun::RecordingStreamBuilder::new("rerun_example_tensor").spawn()?;

    let mut data = Array::<u8, _>::default((8, 6, 3, 5).f());
    let mut rng = rand::rngs::SmallRng::seed_from_u64(42);
    data.map_inplace(|x| *x = rng.gen());

    let tensor =
        rerun::Tensor::try_from(data)?.with_dim_names(["width", "height", "channel", "batch"]);
    rec.log("tensor", &tensor)?;

    Ok(())
}