একটি সারাংশ একটি মডেলের আর্কিটেকচার সম্পর্কে বিশদ প্রদান করে, যেমন স্তরের ধরন এবং আকার।
নকশা প্রস্তাব এখানে পাওয়া যাবে. এই বাস্তবায়নটি একটি 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]