ملخصات النماذج

يوفر الملخص تفاصيل حول بنية النموذج، مثل أنواع الطبقات والأشكال.

يمكن العثور على مقترح التصميم هنا . يعد هذا التنفيذ قيد التنفيذ، لذا يرجى تقديم مشكلة تتعلق بالتحسينات التي ترغب في رؤيتها أو المشكلات التي تواجهها.

عرض ملخص النموذج

قم بإنشاء جهاز وطراز 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]