def ProductWith

bendtorch-core@0.1.0.0 · src/autograd/tensor_vjp_LAWS.bend:43

import Base
import bendtorch-core@0.1.0.0/src/shape/core.bend as Shape
import bendtorch-core@0.1.0.0/src/storage/LAWS.bend as StorageLaws
import bendtorch-core@0.1.0.0/src/tensor/core.bend as Core
import bendtorch-core@0.1.0.0/src/tensor/elementwise.bend as Elementwise
import bendtorch-core@0.1.0.0/src/tensor/contracts/binary_LAWS.bend as Binary

43# What the transport form of tensor_product_replayed asks of P: P holds for
44# both inputs, replayed with any histories that keep their words, next to the
45# output replayed over +0.0 with any history whose words are `cells`.
46def ProductWith(+operation: Elementwise.Operation, +plan: Shape.BroadcastPlan,
47  +left: Shape.Shape, +left_fill: F32, +left_ops: List<&2, StorageLaws.Op>,
48  +right: Shape.Shape, +right_fill: F32, +right_ops: List<&2, StorageLaws.Op>,
49  -P: (Core.Dynamic & Core.Dynamic) & Result<&2, &1, Core.Error, Core.Dynamic> -> Type) -> Type:
50  Binary.BinaryWith(operation, plan, left, left_fill, left_ops, right, right_fill, right_ops, P)