خلاصه جزئیاتی در مورد معماری یک مدل، مانند انواع لایه ها و اشکال ارائه می دهد.
طرح پیشنهادی را می توانید در اینجا بیابید. این پیادهسازی یک 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]