我正在尝试将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] 删除。
我来说两句