api/org.diffkt.model/TrainableTensor

TrainableTensor

class TrainableTensor(tensor: DTensor) : Trainable<TrainableTensor>

Types

Companion

object Companion

Functions

cpu

open override fun cpu(): TrainableTensor

equals

open operator override fun equals(other: Any?): Boolean

extractTangent

open override fun extractTangent(output: DTensor, extractor: (input: DTensor, output: DTensor) -> DTensor): TrainableTensor.Companion.Tangent

gpu

open override fun gpu(): TrainableTensor

hashCode

open override fun hashCode(): Int

load

open override fun load(from: ByteBuffer): TrainableTensor

store

open override fun store(into: ByteBuffer): ByteBuffer

to

open fun to(device: Device): OnDevice

trainingStep

open override fun trainingStep(optim: Optimizer<*>, tangent: Trainable.Tangent): TrainableTensor

wrap

open override fun wrap(wrapper: Wrapper): TrainableTensor

The wrap function should return the same static type it is declared on.

Properties

tensor

val tensor: DTensor