Class: Gsplat::Ops::Exp

Inherits:

Autograd::Function

• Object
• Autograd::Function
• Gsplat::Ops::Exp

Defined in:

lib/gsplat/ops/tensor_value_ops.rb

Overview

Elementwise exponential activation.

Class Method Summary

• .backward(context, gradient) ⇒ Object
• .forward(context, value) ⇒ Object

Methods inherited from Autograd::Function

apply

Class Method Details

.backward(context, gradient) ⇒ Object

# File 'lib/gsplat/ops/tensor_value_ops.rb', line 52

def backward(context, gradient)
  [gradient * context.saved_values.first]
end

.forward(context, value) ⇒ Object

# File 'lib/gsplat/ops/tensor_value_ops.rb', line 46

def forward(context, value)
  output = Numo::NMath.exp(value)
  context.save(output)
  output
end