Package-level declarations

Types

Link copied to clipboard
data class NodeSummary(val name: String, val input: Shape, val output: Shape, val params: Long)
Link copied to clipboard
class Summary<T : DType, V>(ctx: ExecutionContext)

Functions

Link copied to clipboard
fun <T : DType, V> Module<T, V>.describe(input: Shape, ctx: ExecutionContext, dtypeClass: KClass<T>, batch_size: Int = -1): String