org.bytedeco.pytorch

Class FlattenImpl

java.lang.Object

org.bytedeco.javacpp.Pointer

org.bytedeco.pytorch.Module

org.bytedeco.pytorch.FlattenImplCloneable

org.bytedeco.pytorch.FlattenImpl

All Implemented Interfaces:

AutoCloseable

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

A placeholder for Flatten operator See https://pytorch.org/docs/master/generated/torch.nn.Flatten.html to learn about the exact behavior of this module. See the documentation for torch::nn::FlattenOptions class to learn what constructor arguments are supported for this module. Example:

  Flatten model(FlattenOptions().start_dim(2).end_dim(4));
  

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
FlattenImpl()
FlattenImpl(FlattenOptions options_)
FlattenImpl(Module pointer) Downcast constructor.
FlattenImpl(Pointer p) Pointer cast constructor.

Method Summary

Modifier and Type	Method and Description
Tensor	forward(Tensor input) Applies a flatten transform on the input.
FlattenOptions	options() The options used to configure this module.
FlattenImpl	options(FlattenOptions setter)
void	pretty_print(Pointer stream) Pretty prints the Flatten 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.FlattenImplCloneable

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

FlattenImpl

public FlattenImpl(Pointer p)

Pointer cast constructor. Invokes Pointer(Pointer).

FlattenImpl

public FlattenImpl(Module pointer)

Downcast constructor.

FlattenImpl

public FlattenImpl(@Const @ByRef(nullValue="torch::nn::FlattenOptions{}")
                   FlattenOptions options_)

FlattenImpl

public FlattenImpl()

Method Detail

reset

public void reset()

Description copied from class: FlattenImplCloneable

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

Overrides:

reset in class FlattenImplCloneable

pretty_print

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

Pretty prints the Flatten module into the given stream.

Overrides:

pretty_print in class Module

forward

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

Applies a flatten transform on the input.

options

@ByRef
public FlattenOptions options()

The options used to configure this module.

options

public FlattenImpl options(FlattenOptions setter)