SKaiNET
Toggle table of contents
0.20.0-SNAPSHOT
common
Target filter
common
Switch theme
Search in API
Skip to content
SKaiNET
skainet-backend-cpu
/
sk.ainet.exec.tensor.ops
/
DefaultCpuOpsBase
/
scaledDotProductAttention
scaled
Dot
Product
Attention
open
override
fun
<
T
:
DType
,
V
>
scaledDotProductAttention
(
query
:
Tensor
<
T
,
V
>
,
key
:
Tensor
<
T
,
V
>
,
value
:
Tensor
<
T
,
V
>
,
mask
:
Tensor
<
T
,
V
>
?
,
scale
:
Float
,
causal
:
Boolean
)
:
Tensor
<
T
,
V
>
(
source
)