org.bytedeco.pytorch

Class RNNCellImpl

java.lang.Object

org.bytedeco.javacpp.Pointer

org.bytedeco.pytorch.Module

org.bytedeco.pytorch.RNNCellImplCloneable

org.bytedeco.pytorch.RNNCellImplBase

org.bytedeco.pytorch.RNNCellImpl

All Implemented Interfaces:

AutoCloseable

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

An Elman RNN cell with tanh or ReLU non-linearity. See https://pytorch.org/docs/main/nn.html#torch.nn.RNNCell to learn about the exact behavior of this module. See the documentation for torch::nn::RNNCellOptions class to learn what constructor arguments are supported for this module. Example:

  RNNCell model(RNNCellOptions(20,
  10).bias(false).nonlinearity(torch::kReLU));
  

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
RNNCellImpl(long input_size, long hidden_size)
RNNCellImpl(Pointer p) Pointer cast constructor.
RNNCellImpl(RNNCellOptions options_)

Method Summary

Modifier and Type	Method and Description
Tensor	forward(Tensor input)
Tensor	forward(Tensor input, Tensor hx)
RNNCellOptions	options()
RNNCellImpl	options(RNNCellOptions setter)

Methods inherited from class org.bytedeco.pytorch.RNNCellImplBase

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.RNNCellImplCloneable

clone, clone

Methods inherited from class org.bytedeco.pytorch.Module

apply, apply, apply, apply, apply, apply, apply, apply, asAdaptiveAvgPool1d, asAdaptiveAvgPool2d, asAdaptiveAvgPool3d, asAdaptiveLogSoftmaxWithLoss, asAdaptiveMaxPool1d, asAdaptiveMaxPool2d, asAdaptiveMaxPool3d, asAlphaDropout, asAvgPool1d, asAvgPool2d, asAvgPool3d, asBatchNorm1d, asBatchNorm2d, asBatchNorm3d, asBCELoss, asBCEWithLogitsLoss, asBilinear, asCELU, asConstantPad1d, asConstantPad2d, asConstantPad3d, asConv1d, asConv2d, asConv3d, asConvTranspose1d, asConvTranspose2d, asConvTranspose3d, asCosineEmbeddingLoss, asCosineSimilarity, asCrossEntropyLoss, asCrossMapLRN2d, asCTCLoss, asDropout, asDropout2d, asDropout3d, asELU, asEmbedding, asEmbeddingBag, asFeatureAlphaDropout, asFlatten, asFold, asFractionalMaxPool2d, asFractionalMaxPool3d, asGELU, asGLU, asGroupNorm, asGRU, asGRUCell, asHardshrink, asHardtanh, asHingeEmbeddingLoss, asHuberLoss, asIdentity, asInstanceNorm1d, asInstanceNorm2d, asInstanceNorm3d, asKLDivLoss, asL1Loss, asLayerNorm, asLeakyReLU, asLinear, asLocalResponseNorm, asLogSigmoid, asLogSoftmax, asLPPool1d, asLPPool2d, asLPPool3d, asLSTM, asLSTMCell, asMarginRankingLoss, asMaxPool1d, asMaxPool2d, asMaxPool3d, asMaxUnpool1d, asMaxUnpool2d, asMaxUnpool3d, asMish, asModuleDict, asModuleList, asMSELoss, asMultiheadAttention, asMultiLabelMarginLoss, asMultiLabelSoftMarginLoss, asMultiMarginLoss, asNLLLoss, asPairwiseDistance, asParameterDict, asParameterList, asPixelShuffle, asPixelUnshuffle, asPoissonNLLLoss, asPReLU, asReflectionPad1d, asReflectionPad2d, asReflectionPad3d, asReLU, asReLU6, asReplicationPad1d, asReplicationPad2d, asReplicationPad3d, asRNN, asRNNCell, asRReLU, asSELU, asSequential, asSigmoid, asSiLU, asSmoothL1Loss, asSoftMarginLoss, asSoftmax, asSoftmax2d, asSoftmin, asSoftplus, asSoftshrink, asSoftsign, asTanh, asTanhshrink, asThreshold, asTransformer, asTransformerDecoder, asTransformerDecoderLayer, asTransformerEncoder, asTransformerEncoderLayer, asTripletMarginLoss, asTripletMarginWithDistanceLoss, asUnflatten, asUnfold, asUpsample, asZeroPad1d, asZeroPad2d, asZeroPad3d, 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, put, 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

RNNCellImpl

public RNNCellImpl(Pointer p)

Pointer cast constructor. Invokes Pointer(Pointer).

RNNCellImpl

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

RNNCellImpl

public RNNCellImpl(@Const @ByRef
                   RNNCellOptions 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 RNNCellOptions options()

options

public RNNCellImpl options(RNNCellOptions setter)