在Swift中为Tensorflow创建随机的Tensor

Question

我正在尝试将Swift用于Tensorflow API，但我找不到能够随机随机初始化的创建Tensors的方法。

目前，我正在从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])