סיכומי דגם

סיכום מספק פרטים על הארכיטקטורה של מודל, כגון סוגי שכבות וצורות.

את הצעת העיצוב ניתן למצוא כאן . יישום זה הוא WIP, אז בבקשה שלח בעיה עם שיפורים שאתה רוצה לראות או בעיות שאתה נתקל בהן.

צפייה בתקציר הדגם

צור מכשיר ודגם X10.

import TensorFlow

public struct MyModel: Layer {
  public var dense1 = Dense<Float>(inputSize: 1, outputSize: 1)
  public var dense2 = Dense<Float>(inputSize: 4, outputSize: 4)
  public var dense3 = Dense<Float>(inputSize: 4, outputSize: 4)
  public var flatten = Flatten<Float>()

  @differentiable
  public func callAsFunction(_ input: Tensor<Float>) -> Tensor<Float> {
    let layer1 = dense1(input)
    let layer2 = layer1.reshaped(to: [1, 4])
    let layer3 = dense2(layer2)
    let layer4 = dense3(layer3)
    return flatten(layer4)
  }
}

let device = Device.defaultXLA
let model0 = MyModel()
let model = MyModel(copying: model0, to: device)

צור טנזור קלט.

let input = Tensor<Float>(repeating: 1, shape: [1, 4, 1, 1], on: device)

צור סיכום של הדגם שלך.

let summary = model.summary(input: input)
print(summary)

Layer                           Output Shape         Attributes
=============================== ==================== ======================
Dense<Float>                    [1, 4, 1, 1]
Dense<Float>                    [1, 4]
Dense<Float>                    [1, 4]
Flatten<Float>                  [1, 4]