@Namespace(value="torch::nn") @NoOffset @Properties(inherit=torch.class) public class FlattenImpl extends FlattenImplCloneable
torch::nn::FlattenOptions
class to learn what
constructor arguments are supported for this module.
Example:
Flatten model(FlattenOptions().start_dim(2).end_dim(4));
Pointer.CustomDeallocator, Pointer.Deallocator, Pointer.NativeDeallocator, Pointer.ReferenceCounter
Constructor and Description |
---|
FlattenImpl() |
FlattenImpl(FlattenOptions options_) |
FlattenImpl(Module pointer)
Downcast constructor.
|
FlattenImpl(Pointer p)
Pointer cast constructor.
|
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. |
asModule, asModule, clone, clone
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
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
public FlattenImpl(Pointer p)
Pointer(Pointer)
.public FlattenImpl(Module pointer)
public FlattenImpl(@Const @ByRef(nullValue="torch::nn::FlattenOptions{}") FlattenOptions options_)
public FlattenImpl()
public void reset()
FlattenImplCloneable
reset()
must perform initialization of all members with reference
semantics, most importantly parameters, buffers and submodules.reset
in class FlattenImplCloneable
public void pretty_print(@Cast(value="std::ostream*") @ByRef Pointer stream)
Flatten
module into the given stream
.pretty_print
in class Module
@ByVal public Tensor forward(@Const @ByRef Tensor input)
input
.@ByRef public FlattenOptions options()
public FlattenImpl options(FlattenOptions setter)
Copyright © 2024. All rights reserved.