| Package | Description |
|---|---|
| org.bytedeco.pytorch |
| Modifier and Type | Method and Description |
|---|---|
AnyValue |
AnyModule.any_forward(AnyValue input)
Invokes
forward() on the contained module with the given arguments, and
returns the return value as an AnyValue. |
AnyValue |
AnyModule.any_forward(Tensor input) |
AnyValue |
AnyModule.any_forward(Tensor input,
long... output_size) |
AnyValue |
AnyModule.any_forward(Tensor input,
LongArrayRefOptional output_size) |
AnyValue |
AnyModule.any_forward(Tensor input,
T_TensorTensor_TOptional hx_opt) |
AnyValue |
AnyModule.any_forward(Tensor input1,
Tensor input2) |
AnyValue |
AnyModule.any_forward(Tensor input,
Tensor indices,
LongVectorOptional output_size) |
AnyValue |
AnyModule.any_forward(Tensor input1,
Tensor input2,
Tensor input3) |
AnyValue |
AnyModule.any_forward(Tensor input1,
Tensor input2,
Tensor input3,
Tensor input4) |
AnyValue |
AnyModule.any_forward(Tensor query,
Tensor key,
Tensor value,
Tensor key_padding_mask,
boolean need_weights,
Tensor attn_mask,
boolean average_attn_weights) |
AnyValue |
AnyModule.any_forward(Tensor input1,
Tensor input2,
Tensor input3,
Tensor input4,
Tensor input5,
Tensor input6) |
AnyValue |
AnyModule.any_forward(Tensor input1,
Tensor input2,
Tensor input3,
Tensor input4,
Tensor input5,
Tensor input6,
Tensor input7,
Tensor input8) |
AnyValue |
AnyValue.put(AnyValue arg0) |
| Modifier and Type | Method and Description |
|---|---|
AnyValue |
AnyModule.any_forward(AnyValue input)
Invokes
forward() on the contained module with the given arguments, and
returns the return value as an AnyValue. |
AnyValue |
AnyValue.put(AnyValue arg0) |
| Constructor and Description |
|---|
AnyValue(AnyValue arg0)
Move construction and assignment is allowed, and follows the default
behavior of move for
std::unique_ptr. |
Copyright © 2024. All rights reserved.