在 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)!

Tensorflow 有没有Swift获取随机数组的纯方法？

Answer 1

显然，Tensor 有一个 init 函数，用于构造特定形状和随机值的张量。例如：

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

Creating random Tensors in Swift for Tensorflow