Package | Description |
---|---|
org.bytedeco.pytorch.cuda |
Modifier and Type | Method and Description |
---|---|
DropoutDescriptor |
DropoutDescriptor.getPointer(long i) |
DropoutDescriptor |
DropoutDescriptor.position(long position) |
DropoutDescriptor |
DropoutDescriptor.state(Tensor setter) |
Modifier and Type | Method and Description |
---|---|
void |
RNNDescriptor.set(Pointer handle,
int hidden_size,
int proj_size,
int num_layers,
DropoutDescriptor dropout_desc,
int input_mode,
int bidirectional,
int mode,
int datatype,
int input_type,
int algo,
boolean allow_tf32) |
Copyright © 2024. All rights reserved.