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.