BatchedRoutines.batched_gemm
— Functionbatched_gemm(transA, transB, alpha, A, B)
Batched gemm.
BatchedRoutines.batched_gemm!
— Functionbatched_gemm!(transA, transB, alpha, A, B, beta, C)
Batched gemm!.
BatchedRoutines.batched_gemm
— Methodbatched_gemm(alpha, A, B)
BatchedRoutines.batched_scal
— Functionbatched_scal(s, X)
Return X
scaled by a
for all the batch.
BatchedRoutines.batched_scal!
— Functionbatched_scal!(s, X)
Overwrite X[:, :, i]
with a[i] * X[:, :, i]
, where i
is the batch dimension.
BatchedRoutines.batched_tr
— Methodbatched_tr(A::AbstractArray{T, 3}) where T
Batched version of trace.