lamp-core/lamp.autograd/NllLoss

NllLoss

case class NllLoss(scope: Scope, input: Variable, target: STen, weights: STen, reduction: Reduction, ignore: Long) extends Op
trait Serializable
trait Product
trait Equals
trait Op
class Object
trait Matchable
class Any

Value members

Inherited methods

def productElementNames: Iterator[String]
Inherited from:
Product
def productIterator: Iterator[Any]
Inherited from:
Product

Concrete fields

val params: List[(Variable, (STen, STen) => Unit)]
val total_weight: Tensor
val value1: Tensor