【问题标题】:Creating random Tensors in Swift for Tensorflow在 Swift 中为 Tensorflow 创建随机张量
【发布时间】:2019-04-16 21:41:35
【问题描述】:

我正在尝试将 Swift 用于 Tensorflow API,但我无法找到一种方法来创建具有随机正常初始化的张量。

目前我正在从 numpy 获取一个数组来构造一个张量结构:

let np = Python.import("numpy")

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

是否有 Swift for Tensorflow 纯方式获取随机数组?

【问题讨论】:

    标签: swift numpy tensorflow random tensor


    【解决方案1】:

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

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

    【讨论】:

      猜你喜欢
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      • 2020-10-09
      • 1970-01-01
      • 1970-01-01
      • 2020-10-03
      • 1970-01-01
      • 2018-08-08
      相关资源
      最近更新 更多