type Geometry

bendtorch-core@0.1.0.0 · src/tensor/batched_matmul_LAWS.bend:386

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

386type Geometry<-output: List<&2, Nat>, -left: List<&2, Nat>, -right: List<&2, Nat>, -batch: List<&2, Nat>, -vectors: Bool & Bool,
387  -plan: Batched.Plan> is Data:
388  Geometry{-output_dims: {Shape.naturals(Shape.dimensions(Batched.output_of(plan))) == output : List<&2, Nat>},
389    -left_dims: {left_view(plan) == left : List<&2, Nat>},
390    -right_dims: {right_view(plan) == right : List<&2, Nat>},
391    -batch_dims: {batch_view(plan) == batch : List<&2, Nat>},
392    -vector_dims: {vector_view(plan) == vectors : Bool & Bool}}