org.bytedeco.pytorch

Class UnfoldImpl

java.lang.Object

org.bytedeco.javacpp.Pointer

org.bytedeco.pytorch.Module

org.bytedeco.pytorch.UnfoldImplCloneable

org.bytedeco.pytorch.UnfoldImpl

All Implemented Interfaces:

AutoCloseable

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

Applies unfold over a 4-D input. See https://pytorch.org/docs/master/nn.html#torch.nn.Unfold to learn about the exact behavior of this module. See the documentation for torch::nn::UnfoldOptions class to learn what constructor arguments are supported for this module. Example:

  Unfold model(UnfoldOptions({2, 4}).dilation(2).padding({2, 1}).stride(2));
  

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
UnfoldImpl(LongPointer kernel_size)
UnfoldImpl(Module pointer) Downcast constructor.
UnfoldImpl(Pointer p) Pointer cast constructor.
UnfoldImpl(UnfoldOptions options_)

Method Summary

Modifier and Type	Method and Description
Tensor	forward(Tensor input)
UnfoldOptions	options() The options with which this Module was constructed.
UnfoldImpl	options(UnfoldOptions setter)
void	pretty_print(Pointer stream) Pretty prints the Unfold module into the given stream.
void	reset() reset() must perform initialization of all members with reference semantics, most importantly parameters, buffers and submodules.

Methods inherited from class org.bytedeco.pytorch.UnfoldImplCloneable

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

UnfoldImpl

public UnfoldImpl(Pointer p)

Pointer cast constructor. Invokes Pointer(Pointer).

UnfoldImpl

public UnfoldImpl(Module pointer)

Downcast constructor.

UnfoldImpl

public UnfoldImpl(@ByVal @Cast(value="torch::ExpandingArray<2>*")
                  LongPointer kernel_size)

UnfoldImpl

public UnfoldImpl(@Const @ByRef
                  UnfoldOptions options_)

Method Detail

reset

public void reset()

Description copied from class: UnfoldImplCloneable

reset() must perform initialization of all members with reference semantics, most importantly parameters, buffers and submodules.

Overrides:

reset in class UnfoldImplCloneable

pretty_print

public void pretty_print(@Cast(value="std::ostream*") @ByRef
                         Pointer stream)

Pretty prints the Unfold module into the given stream.

Overrides:

pretty_print in class Module

forward

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

options

@ByRef
public UnfoldOptions options()

The options with which this Module was constructed.

options

public UnfoldImpl options(UnfoldOptions setter)