Args

class Args<T : DType, V>(feed: Map<String, Tensor<T, V>>)(source)

Arguments accessor passed to the compute lambda. Retrieves tensors by input name.

Constructors

Link copied to clipboard
constructor(feed: Map<String, Tensor<T, V>>)

Functions

Link copied to clipboard
operator fun get(name: String): Tensor<T, V>