def apply

bendtorch-core@0.1.0.0 · src/tensor/transform_observation.bend:11

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
import bendtorch-core@0.1.0.0/src/tensor/transforms.bend as Transforms

11def apply(+input: Shape.Shape, tensor: Core.Tensor<input>, request: Request) ->
12  Core.Tensor<input> & Result<&2, &1, Transforms.Error, Core.Dynamic>:
13  match request:
14    case Permute{+axes}:
15      Transforms.permute_copy(input, tensor, axes)
16    case Transpose{+a, +b}:
17      Transforms.transpose_copy(input, tensor, a, b)