| Package | Description |
|---|---|
| org.bytedeco.pytorch | |
| org.bytedeco.pytorch.global |
| Modifier and Type | Method and Description |
|---|---|
T_TensorTensorVectorTensorVector_T |
T_TensorTensorVectorTensorVector_T.put(T_TensorTensorVectorTensorVector_T x) |
| Modifier and Type | Method and Description |
|---|---|
static Tensor |
T_TensorTensorVectorTensorVector_T.get0(T_TensorTensorVectorTensorVector_T container) |
static TensorVector |
T_TensorTensorVectorTensorVector_T.get1(T_TensorTensorVectorTensorVector_T container) |
static TensorVector |
T_TensorTensorVectorTensorVector_T.get2(T_TensorTensorVectorTensorVector_T container) |
T_TensorTensorVectorTensorVector_T |
T_TensorTensorVectorTensorVector_T.put(T_TensorTensorVectorTensorVector_T x) |
| Modifier and Type | Method and Description |
|---|---|
static T_TensorTensorVectorTensorVector_T |
torch.lstm_mps_backward(TensorOptional grad_y,
TensorOptional grad_hy,
TensorOptional grad_cy,
Tensor z_state,
Tensor cell_state_fwd,
Tensor input,
Tensor layersOutputs,
TensorArrayRef hx,
TensorArrayRef params,
boolean has_biases,
long num_layers,
double dropout,
boolean train,
boolean bidirectional,
boolean batch_first) |
static T_TensorTensorVectorTensorVector_T |
torch.lstm_mps_backward(TensorOptional grad_y,
TensorOptional grad_hy,
TensorOptional grad_cy,
Tensor z_state,
Tensor cell_state_fwd,
Tensor input,
Tensor layersOutputs,
TensorVector hx,
TensorVector params,
boolean has_biases,
long num_layers,
double dropout,
boolean train,
boolean bidirectional,
boolean batch_first) |
Copyright © 2024. All rights reserved.