def relabeled
bendtorch-core@0.1.0.0 · src/autograd/tensor_backward.bend:288
import bendtorch-core@0.1.0.0/src/autograd/tensor_backward.bend as TensorBackward
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
288
289
290def relabeled(+shape: Shape.Shape, gradient: Core.Dynamic) -> Core.Dynamic:
291 Core.Packed{+held, tensor} = gradient
292 relabel_decided(held, shape, tensor, Shape.bool_case(Nat.is_eq(Shape.elements(held), Shape.elements(shape))))