在Swift中为Tensorflow创建随机的Tensor

巴赫尔

我正在尝试将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的纯方法来获取随机数组?

巴赫尔

显然,Tensor具有init构造一定形状和随机值的张量功能。例如:

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

本文收集自互联网,转载请注明来源。

如有侵权,请联系 [email protected] 删除。

编辑于
0

我来说两句

0 条评论
登录 后参与评论

相关文章