Function tile_device

pub fn tile_device(
    a: impl AsRef<Array>,
    reps: &[i32],
    stream: impl AsRef<Stream>,
) -> Result<Array>

Construct an array by repeating a the number of times given by reps.

Params

• a: The input array.
• reps: The number of repetitions along each axis.

Example

use mlx_rs::{Array, ops::*};

let x = Array::from_slice(&[1, 2, 3], &[3]);
let y = tile(&x, &[2]);