batchDot

fun batchDot(scope: Scope?, x: Operand<Float>, y: Operand<Float>, axis: IntArray): Operand<Float>
Content copied to clipboard

Batchwise dot product.

Parameters

scope

: Current scope

x

: Operand with dimensions>=2

y

: Operand with dimensions>=2

axis

: Axis along which to perform batch dot