nn.gates.loss

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

bce_gate exposes this operation as part of the public API.

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

huber_loss_gate exposes this operation as part of the public API.

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

kl_div_loss_gate exposes this operation as part of the public API.

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

mse_gate exposes this operation as part of the public API.

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

nll_loss_gate exposes this operation as part of the public API.

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

sigmoid_cross_entropy_gate exposes this operation as part of the public API.

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

softmax_cross_entropy_gate exposes this operation as part of the public API.

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

backward exposes this operation as part of the public API.

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

cache exposes this operation as part of the public API.

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

backward exposes this operation as part of the public API.

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

cache exposes this operation as part of the public API.

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

backward exposes this operation as part of the public API.

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

cache exposes this operation as part of the public API.

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

backward exposes this operation as part of the public API.

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

cache exposes this operation as part of the public API.

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

backward exposes this operation as part of the public API.

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

cache exposes this operation as part of the public API.

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

backward exposes this operation as part of the public API.

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

cache exposes this operation as part of the public API.

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

backward exposes this operation as part of the public API.

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

cache exposes this operation as part of the public API.

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

BCEGate defines a public data structure for this module.

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

HuberLossGate defines a public data structure for this module.

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

KLDivLossGate defines a public data structure for this module.

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

MseGate defines a public data structure for this module.

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

NLLLossGate defines a public data structure for this module.

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

SigmoidCrossEntropyGate defines a public data structure for this module.

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

SoftmaxCrossEntropyGate defines a public data structure for this module.