mlx_rs::ops

Function tile

Source
pub fn tile(a: impl AsRef<Array>, reps: &[i32]) -> Result<Array>
Expand description

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]);