import numpy as np
from numpy.typing import NDArray
import tempfile
import mlflow
import os

def initialise_random_tensor(*dims: int) -> NDArray[np.float32]:
    return np.random.rand(*dims).astype(np.float32)