Создание случайных тензоров в Swift для Tensorflow

Question

Я пытаюсь использовать Swift для Tensorflow API и не могу найти способ создания тензоров со случайной нормальной инициализацией.

В настоящее время я получаю массив из numpy для созданияСтруктура Tensor:

let np = Python.import("numpy")

let w_np = np.random.randn(10, 10).astype(np.float32)
let w = Tensor<Float32>(numpy: w_np)!

Существует ли Swift для Tensorflow - чистый способ получения случайных массивов?

Answer 1

Очевидно, Tensor имеет функцию init для построения тензора определенной формы и случайных значений.Например:

let input = Tensor<Float>(randomUniform: [64, 28, 28, 32])