| Package | Description |
|---|---|
| org.bytedeco.pytorch.global |
| Modifier and Type | Method and Description |
|---|---|
static T_TensorTensor_T |
torch.multi_head_attention_forward(Tensor query,
Tensor key,
Tensor value,
MultiheadAttentionForwardFuncOptions options) |
Copyright © 2024. All rights reserved.