| Package | Description |
|---|---|
| org.bytedeco.pytorch | |
| org.bytedeco.pytorch.global |
| Modifier and Type | Method and Description |
|---|---|
T_TensorTensorTensorTensorVector_T |
T_TensorTensorTensorTensorVector_T.put(T_TensorTensorTensorTensorVector_T x) |
| Modifier and Type | Method and Description |
|---|---|
static Tensor |
T_TensorTensorTensorTensorVector_T.get0(T_TensorTensorTensorTensorVector_T container) |
static Tensor |
T_TensorTensorTensorTensorVector_T.get1(T_TensorTensorTensorTensorVector_T container) |
static Tensor |
T_TensorTensorTensorTensorVector_T.get2(T_TensorTensorTensorTensorVector_T container) |
static TensorVector |
T_TensorTensorTensorTensorVector_T.get3(T_TensorTensorTensorTensorVector_T container) |
T_TensorTensorTensorTensorVector_T |
T_TensorTensorTensorTensorVector_T.put(T_TensorTensorTensorTensorVector_T x) |
| Modifier and Type | Method and Description |
|---|---|
static T_TensorTensorTensorTensorVector_T |
torch.miopen_rnn_backward(Tensor input,
TensorArrayRef weight,
long weight_stride0,
Tensor weight_buf,
Tensor hx,
TensorOptional cx,
Tensor output,
TensorOptional grad_output,
TensorOptional grad_hy,
TensorOptional grad_cy,
long mode,
long hidden_size,
long num_layers,
boolean batch_first,
double dropout,
boolean train,
boolean bidirectional,
LongArrayRef batch_sizes,
TensorOptional dropout_state,
Tensor reserve,
BoolPointer output_mask) |
static T_TensorTensorTensorTensorVector_T |
torch.miopen_rnn_backward(Tensor input,
TensorVector weight,
long weight_stride0,
Tensor weight_buf,
Tensor hx,
TensorOptional cx,
Tensor output,
TensorOptional grad_output,
TensorOptional grad_hy,
TensorOptional grad_cy,
long mode,
long hidden_size,
long num_layers,
boolean batch_first,
double dropout,
boolean train,
boolean bidirectional,
long[] batch_sizes,
TensorOptional dropout_state,
Tensor reserve,
BoolPointer output_mask) |
Copyright © 2024. All rights reserved.