org.bytedeco.pytorch

Class GRUCellImpl

java.lang.Object

org.bytedeco.javacpp.Pointer

org.bytedeco.pytorch.Module

org.bytedeco.pytorch.GRUCellImplCloneable

org.bytedeco.pytorch.GRUCellImplBase

org.bytedeco.pytorch.GRUCellImpl

All Implemented Interfaces:

AutoCloseable

@Namespace(value="torch::nn")
 @NoOffset
 @Properties(inherit=torch.class)
public class GRUCellImpl
extends GRUCellImplBase

A gated recurrent unit (GRU) cell. See https://pytorch.org/docs/master/nn.html#torch.nn.GRUCell to learn about the exact behavior of this module. See the documentation for torch::nn::GRUCellOptions class to learn what constructor arguments are supported for this module. Example:

  GRUCell model(GRUCellOptions(20, 10).bias(false));
  

Nested Class Summary

Nested classes/interfaces inherited from class org.bytedeco.javacpp.Pointer

Pointer.CustomDeallocator, Pointer.Deallocator, Pointer.NativeDeallocator, Pointer.ReferenceCounter

Field Summary

Fields inherited from class org.bytedeco.javacpp.Pointer

address, capacity, limit, position

Constructor Summary

Constructors

Constructor and Description
GRUCellImpl(GRUCellOptions options_)
GRUCellImpl(long input_size, long hidden_size)
GRUCellImpl(Module pointer) Downcast constructor.
GRUCellImpl(Pointer p) Pointer cast constructor.

Method Summary

Modifier and Type	Method and Description
Tensor	forward(Tensor input)
Tensor	forward(Tensor input, Tensor hx)
GRUCellOptions	options()
GRUCellImpl	options(GRUCellOptions setter)

Methods inherited from class org.bytedeco.pytorch.GRUCellImplBase

bias_hh, bias_hh, bias_ih, bias_ih, options_base, options_base, pretty_print, reset_parameters, reset, weight_hh, weight_hh, weight_ih, weight_ih

Methods inherited from class org.bytedeco.pytorch.GRUCellImplCloneable

asModule, asModule, clone, clone

Methods inherited from class org.bytedeco.pytorch.Module

apply, apply, apply, apply, apply, apply, apply, apply, buffers, buffers, children, eval, is_serializable, is_training, load, modules, modules, name, named_buffers, named_buffers, named_children, named_modules, named_modules, named_modules, named_parameters, named_parameters, parameters, parameters, register_buffer, register_buffer, register_module, register_module, register_parameter, register_parameter, register_parameter, register_parameter, save, shiftLeft, to, to, to, train, unregister_module, unregister_module, zero_grad

Methods inherited from class org.bytedeco.javacpp.Pointer

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

Methods inherited from class java.lang.Object

finalize, getClass, notify, notifyAll, wait, wait, wait

Constructor Detail

GRUCellImpl

public GRUCellImpl(Pointer p)

Pointer cast constructor. Invokes Pointer(Pointer).

GRUCellImpl

public GRUCellImpl(Module pointer)

Downcast constructor.

GRUCellImpl

public GRUCellImpl(@Cast(value="int64_t")
                   long input_size,
                   @Cast(value="int64_t")
                   long hidden_size)

GRUCellImpl

public GRUCellImpl(@Const @ByRef
                   GRUCellOptions options_)

Method Detail

forward

@ByVal
public Tensor forward(@Const @ByRef
                             Tensor input,
                             @ByVal(nullValue="torch::Tensor{}")
                             Tensor hx)

forward

@ByVal
public Tensor forward(@Const @ByRef
                             Tensor input)

options

@ByRef
public GRUCellOptions options()

options

public GRUCellImpl options(GRUCellOptions setter)