Package org.jetbrains.kotlinx.dl.api.core.summary

Types

LayerSummary
Link copied to clipboard
data class LayerSummary(name: String, type: String, outputShape: TensorShape, paramsCount: Long, inboundLayers: List<String>)

The common information about layer.

TfModelSummary
Link copied to clipboard
data class TfModelSummary(type: String, name: String?, layersSummaries: List<LayerSummary>, trainableParamsCount: Long, frozenParamsCount: Long) : ModelSummary

The common information about model.