def reshaped

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

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

284def reshaped(+dimensions: List<&2, U32>, pair: Core.Dynamic & Result<&2, &1, Slice.Error, Core.Dynamic>) ->
285  Core.Dynamic & Result<&2, &1, Slice.Error, Core.Dynamic>:
286  (input, result) = pair
287  match result:
288    case Fail{error}:
289      (input, Fail{error})
290    case Done{output}:
291      (input, reshape_as_slice(Transforms.reshape_dynamic(output, dimensions)))