nn.gates.loss

fn bce_gate[T](input &vtl.Tensor[T], target &vtl.Tensor[T], from_logits bool) &BCEGate[T]

fn huber_loss_gate[T](input &vtl.Tensor[T], target &vtl.Tensor[T], delta T) &HuberLossGate[T]

fn kl_div_loss_gate[T](input &vtl.Tensor[T], target &vtl.Tensor[T]) &KLDivLossGate[T]

fn mse_gate[T](cache &autograd.Variable[T], target &vtl.Tensor[T]) &MseGate[T]

fn nll_loss_gate[T](input &vtl.Tensor[T], target &vtl.Tensor[T]) &NLLLossGate[T]

fn sigmoid_cross_entropy_gate[T](cache &autograd.Variable[T], target &vtl.Tensor[T]) &SigmoidCrossEntropyGate[T]

fn softmax_cross_entropy_gate[T](cache &autograd.Variable[T], target &vtl.Tensor[T]) &SoftmaxCrossEntropyGate[T]

fn (g &BCEGate[T]) backward[T](payload &autograd.Payload[T]) ![]&vtl.Tensor[T]

fn (g &BCEGate[T]) cache[T](mut result autograd.Variable[T], args ...autograd.CacheParam) !

fn (g &HuberLossGate[T]) backward[T](payload &autograd.Payload[T]) ![]&vtl.Tensor[T]

fn (g &HuberLossGate[T]) cache[T](mut result autograd.Variable[T], args ...autograd.CacheParam) !

fn (g &KLDivLossGate[T]) backward[T](payload &autograd.Payload[T]) ![]&vtl.Tensor[T]

fn (g &KLDivLossGate[T]) cache[T](mut result autograd.Variable[T], args ...autograd.CacheParam) !

fn (g &MseGate[T]) backward[T](payload &autograd.Payload[T]) ![]&vtl.Tensor[T]

fn (g &MseGate[T]) cache[T](mut result autograd.Variable[T], args ...autograd.CacheParam) !

fn (g &NLLLossGate[T]) backward[T](payload &autograd.Payload[T]) ![]&vtl.Tensor[T]

fn (g &NLLLossGate[T]) cache[T](mut result autograd.Variable[T], args ...autograd.CacheParam) !

fn (g &SigmoidCrossEntropyGate[T]) backward[T](payload &autograd.Payload[T]) ![]&vtl.Tensor[T]

fn (g &SigmoidCrossEntropyGate[T]) cache[T](mut result autograd.Variable[T], args ...autograd.CacheParam) !

fn (g &SoftmaxCrossEntropyGate[T]) backward[T](payload &autograd.Payload[T]) ![]&vtl.Tensor[T]

fn (g &SoftmaxCrossEntropyGate[T]) cache[T](mut result autograd.Variable[T], args ...autograd.CacheParam) !

struct BCEGate[T] {
pub:
	target      &vtl.Tensor[T] = unsafe { nil }
	from_logits bool
}

struct HuberLossGate[T] {
pub:
	target &vtl.Tensor[T] = unsafe { nil }
	delta  T
}

struct KLDivLossGate[T] {
pub:
	target &vtl.Tensor[T] = unsafe { nil }
}

struct MseGate[T] {
pub:
	cache  &autograd.Variable[T] = unsafe { nil }
	target &vtl.Tensor[T]        = unsafe { nil }
}

struct NLLLossGate[T] {
pub:
	target &vtl.Tensor[T] = unsafe { nil }
}

struct SigmoidCrossEntropyGate[T] {
pub:
	cache  &autograd.Variable[T] = unsafe { nil }
	target &vtl.Tensor[T]        = unsafe { nil }
}

struct SoftmaxCrossEntropyGate[T] {
pub:
	cache  &autograd.Variable[T] = unsafe { nil }
	target &vtl.Tensor[T]        = unsafe { nil }
}