class

GS::Autograd::MatmulBackward

Inherits GS::Autograd::GradFn < Reference < Object

Gradient function for matrix multiplication: C = A @ B dL/dA = dL/dC @ B^T dL/dB = A^T @ dL/dC

Constructors

new(a_data : Tensor, b_data : Tensor)
Source

Instance methods

backward(grad_output : Tensor) : Array(Tensor | Nil)

Compute gradients w.r.t. inputs given gradient of output Returns array of gradients, one per input (or nil if input doesn't require grad)

Source