~/jendhub

def right_gradient

bendtorch-core@0.1.0.0 · src/tensor/batched_matmul.bend:391

import bendtorch-core@0.1.0.0/src/tensor/batched_matmul.bend as BatchedMatmul
import Base
import bendtorch-core@0.1.0.0/src/shape/core.bend as Shape
import bendtorch-core@0.1.0.0/src/tensor/core.bend as Core

391# The right operand's gradient over the output's batch, left^T @ g, of
392# dimensions `right_dims(plan)`.
393def right_gradient(+plan: Plan, left: Core.Dynamic, g: Core.Dynamic) ->
394  Result<&2, &1, Core.Error, Core.Dynamic & Core.Dynamic & Core.Dynamic>:
395  gradient_shaped(right_layout(plan), left, g, right_dims(plan), Shape.construct(right_dims(plan)))

Uses (9)

Used by (1)