def reshape_as_slice

bendtorch-core@0.1.0.0 · src/tensor/slice_LAWS.bend:274

import Base
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
import bendtorch-core@0.1.0.0/src/tensor/slice.bend as Slice

274# A slice's result with its output reshaped to `dimensions`.
275def reshape_as_slice(result: Result<&2, &1, Transforms.Error, Core.Dynamic>) -> Result<&2, &1, Slice.Error, Core.Dynamic>:
276  match result:
277    case Fail{Transforms.LayoutError{error}}:
278      Fail{Slice.LayoutError{error}}
279    case Fail{Transforms.TensorError{error}}:
280      Fail{Slice.TensorError{error}}
281    case Done{output}:
282      Done{output}