@Namespace(value="torch::nn::functional") @NoOffset @Properties(inherit=torch.class) public class MultiheadAttentionForwardFuncOptions extends Pointer
torch::nn::functional::multi_head_attention_forward
Pointer.CustomDeallocator, Pointer.Deallocator, Pointer.NativeDeallocator, Pointer.ReferenceCounter
Constructor and Description |
---|
MultiheadAttentionForwardFuncOptions(long embed_dim_to_check,
long num_heads,
Tensor in_proj_weight,
Tensor in_proj_bias,
Tensor bias_k,
Tensor bias_v,
boolean add_zero_attn,
double dropout_p,
Tensor out_proj_weight,
Tensor out_proj_bias) |
MultiheadAttentionForwardFuncOptions(Pointer p)
Pointer cast constructor.
|
Modifier and Type | Method and Description |
---|---|
BoolPointer |
add_zero_attn() |
Tensor |
attn_mask() |
BoolPointer |
average_attn_weights() |
Tensor |
bias_k() |
Tensor |
bias_v() |
DoublePointer |
dropout_p() |
LongPointer |
embed_dim_to_check() |
Tensor |
in_proj_bias() |
Tensor |
in_proj_weight() |
Tensor |
k_proj_weight() |
Tensor |
key_padding_mask() |
BoolPointer |
need_weights() |
LongPointer |
num_heads() |
Tensor |
out_proj_bias() |
Tensor |
out_proj_weight() |
Tensor |
q_proj_weight() |
Tensor |
static_k() |
Tensor |
static_v() |
BoolPointer |
training() |
BoolPointer |
use_separate_proj_weight() |
Tensor |
v_proj_weight() |
address, asBuffer, asByteBuffer, availablePhysicalBytes, calloc, capacity, capacity, close, deallocate, deallocate, deallocateReferences, deallocator, deallocator, equals, fill, formatBytes, free, getDirectBufferAddress, getPointer, getPointer, getPointer, getPointer, hashCode, interruptDeallocatorThread, isNull, isNull, limit, limit, malloc, maxBytes, maxPhysicalBytes, memchr, memcmp, memcpy, memmove, memset, offsetAddress, offsetof, offsetof, parseBytes, physicalBytes, physicalBytesInaccurate, position, position, put, realloc, referenceCount, releaseReference, retainReference, setNull, sizeof, sizeof, toString, totalBytes, totalCount, totalPhysicalBytes, withDeallocator, zero
public MultiheadAttentionForwardFuncOptions(Pointer p)
Pointer(Pointer)
.public MultiheadAttentionForwardFuncOptions(@Cast(value="int64_t") long embed_dim_to_check, @Cast(value="int64_t") long num_heads, @ByVal Tensor in_proj_weight, @ByVal Tensor in_proj_bias, @ByVal Tensor bias_k, @ByVal Tensor bias_v, @Cast(value="bool") boolean add_zero_attn, double dropout_p, @ByVal Tensor out_proj_weight, @ByVal Tensor out_proj_bias)
@Cast(value="int64_t*") @ByRef @NoException(value=true) public LongPointer embed_dim_to_check()
@Cast(value="int64_t*") @ByRef @NoException(value=true) public LongPointer num_heads()
@ByRef @NoException(value=true) public Tensor in_proj_weight()
@ByRef @NoException(value=true) public Tensor in_proj_bias()
@ByRef @NoException(value=true) public Tensor bias_k()
@ByRef @NoException(value=true) public Tensor bias_v()
@Cast(value="bool*") @ByRef @NoException(value=true) public BoolPointer add_zero_attn()
@ByRef @NoException(value=true) public DoublePointer dropout_p()
@ByRef @NoException(value=true) public Tensor out_proj_weight()
@ByRef @NoException(value=true) public Tensor out_proj_bias()
@Cast(value="bool*") @ByRef @NoException(value=true) public BoolPointer training()
@ByRef @NoException(value=true) public Tensor key_padding_mask()
@Cast(value="bool*") @ByRef @NoException(value=true) public BoolPointer need_weights()
@ByRef @NoException(value=true) public Tensor attn_mask()
@Cast(value="bool*") @ByRef @NoException(value=true) public BoolPointer use_separate_proj_weight()
@ByRef @NoException(value=true) public Tensor q_proj_weight()
@ByRef @NoException(value=true) public Tensor k_proj_weight()
@ByRef @NoException(value=true) public Tensor v_proj_weight()
@ByRef @NoException(value=true) public Tensor static_k()
@ByRef @NoException(value=true) public Tensor static_v()
@Cast(value="bool*") @ByRef @NoException(value=true) public BoolPointer average_attn_weights()
Copyright © 2024. All rights reserved.