org.bytedeco.pytorch

Class TensorIteratorConfig

java.lang.Object

org.bytedeco.javacpp.Pointer

org.bytedeco.pytorch.TensorIteratorConfig

All Implemented Interfaces:

AutoCloseable

@Namespace(value="at")
 @NoOffset
 @Properties(inherit=torch.class)
public class TensorIteratorConfig
extends Pointer

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
TensorIteratorConfig()
TensorIteratorConfig(long size) Native array allocator.
TensorIteratorConfig(Pointer p) Pointer cast constructor.

Method Summary

Modifier and Type	Method and Description
TensorIteratorConfig	add_borrowed_const_input(TensorBase input)
TensorIteratorConfig	add_borrowed_input(TensorBase input)
TensorIteratorConfig	add_borrowed_output(TensorBase output)
TensorIteratorConfig	add_const_input(TensorBase input)
TensorIteratorConfig	add_input(TensorBase input)
TensorIteratorConfig	add_output(TensorBase output) Construction
TensorIteratorConfig	add_owned_const_input(TensorBase input)
TensorIteratorConfig	add_owned_input(TensorBase input)
TensorIteratorConfig	add_owned_output(TensorBase output)
TensorIteratorConfig	allow_cpu_scalars(boolean _allow_cpu_scalars)
TensorIterator	build()
TensorIteratorConfig	cast_common_dtype_to_outputs(boolean _cast_common_dtype_to_outputs)
TensorIteratorConfig	check_all_same_device(boolean _check_all_same_device)
TensorIteratorConfig	check_all_same_dtype(boolean _check_all_same_dtype)
TensorIteratorConfig	declare_static_device(Device device)
TensorIteratorConfig	declare_static_dtype_and_device(torch.ScalarType dtype, Device device)
TensorIteratorConfig	declare_static_dtype(torch.ScalarType dtype)
TensorIteratorConfig	declare_static_shape(long... shape)
TensorIteratorConfig	declare_static_shape(long[] shape, long... squash_dims)
TensorIteratorConfig	declare_static_shape(LongArrayRef shape)
TensorIteratorConfig	declare_static_shape(LongArrayRef shape, LongArrayRef squash_dims)
TensorIteratorConfig	enforce_linear_iteration()
TensorIteratorConfig	enforce_linear_iteration(boolean _enforce_linear_iteration)
TensorIteratorConfig	enforce_safe_casting_to_output(boolean _enforce_safe_casting_to_output)
TensorIteratorConfig	getPointer(long i)
TensorIteratorConfig	is_reduction(boolean _is_reduction)
TensorIteratorConfig	position(long position)
TensorIteratorConfig	promote_inputs_to_common_dtype(boolean _promote_inputs_to_common_dtype)
TensorIteratorConfig	promote_integer_inputs_to_float(boolean _promote_integer_inputs_to_float)
TensorIteratorConfig	resize_outputs(boolean resize_outputs)
TensorIteratorConfig	set_check_mem_overlap(boolean check_mem_overlap)

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, hashCode, interruptDeallocatorThread, isNull, isNull, limit, limit, malloc, maxBytes, maxPhysicalBytes, memchr, memcmp, memcpy, memmove, memset, offsetAddress, offsetof, offsetof, parseBytes, physicalBytes, physicalBytesInaccurate, position, put, realloc, referenceCount, releaseReference, retainReference, setNull, sizeof, sizeof, toString, totalBytes, totalCount, totalPhysicalBytes, withDeallocator, zero

Methods inherited from class java.lang.Object

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

Constructor Detail

TensorIteratorConfig

public TensorIteratorConfig(Pointer p)

Pointer cast constructor. Invokes Pointer(Pointer).

TensorIteratorConfig

public TensorIteratorConfig(long size)

Native array allocator. Access with Pointer.position(long).

TensorIteratorConfig

public TensorIteratorConfig()

Method Detail

position

public TensorIteratorConfig position(long position)

Overrides:

position in class Pointer

getPointer

public TensorIteratorConfig getPointer(long i)

Overrides:

getPointer in class Pointer

add_output

@ByRef
public TensorIteratorConfig add_output(@Const @ByRef
                                              TensorBase output)

Construction

add_input

@ByRef
public TensorIteratorConfig add_input(@Const @ByRef
                                             TensorBase input)

add_const_input

@ByRef
public TensorIteratorConfig add_const_input(@Const @ByRef
                                                   TensorBase input)

add_owned_output

@ByRef
public TensorIteratorConfig add_owned_output(@Const @ByRef
                                                    TensorBase output)

add_owned_input

@ByRef
public TensorIteratorConfig add_owned_input(@Const @ByRef
                                                   TensorBase input)

add_owned_const_input

@ByRef
public TensorIteratorConfig add_owned_const_input(@Const @ByRef
                                                         TensorBase input)

add_borrowed_output

@ByRef
public TensorIteratorConfig add_borrowed_output(@Const @ByRef
                                                       TensorBase output)

add_borrowed_input

@ByRef
public TensorIteratorConfig add_borrowed_input(@Const @ByRef
                                                      TensorBase input)

add_borrowed_const_input

@ByRef
public TensorIteratorConfig add_borrowed_const_input(@Const @ByRef
                                                            TensorBase input)

set_check_mem_overlap

@ByRef
public TensorIteratorConfig set_check_mem_overlap(@Cast(value="bool")
                                                         boolean check_mem_overlap)

check_all_same_dtype

@ByRef
public TensorIteratorConfig check_all_same_dtype(@Cast(value="const bool")
                                                        boolean _check_all_same_dtype)

check_all_same_device

@ByRef
public TensorIteratorConfig check_all_same_device(@Cast(value="const bool")
                                                         boolean _check_all_same_device)

enforce_safe_casting_to_output

@ByRef
public TensorIteratorConfig enforce_safe_casting_to_output(@Cast(value="const bool")
                                                                  boolean _enforce_safe_casting_to_output)

enforce_linear_iteration

@ByRef
public TensorIteratorConfig enforce_linear_iteration(@Cast(value="const bool")
                                                            boolean _enforce_linear_iteration)

enforce_linear_iteration

@ByRef
public TensorIteratorConfig enforce_linear_iteration()

promote_inputs_to_common_dtype

@ByRef
public TensorIteratorConfig promote_inputs_to_common_dtype(@Cast(value="const bool")
                                                                  boolean _promote_inputs_to_common_dtype)

promote_integer_inputs_to_float

@ByRef
public TensorIteratorConfig promote_integer_inputs_to_float(@Cast(value="const bool")
                                                                   boolean _promote_integer_inputs_to_float)

is_reduction

@ByRef
public TensorIteratorConfig is_reduction(@Cast(value="const bool")
                                                boolean _is_reduction)

allow_cpu_scalars

@ByRef
public TensorIteratorConfig allow_cpu_scalars(@Cast(value="const bool")
                                                     boolean _allow_cpu_scalars)

cast_common_dtype_to_outputs

@ByRef
public TensorIteratorConfig cast_common_dtype_to_outputs(@Cast(value="const bool")
                                                                boolean _cast_common_dtype_to_outputs)

resize_outputs

@ByRef
public TensorIteratorConfig resize_outputs(@Cast(value="bool")
                                                  boolean resize_outputs)

declare_static_dtype_and_device

@ByRef
public TensorIteratorConfig declare_static_dtype_and_device(torch.ScalarType dtype,
                                                                   @ByVal
                                                                   Device device)

declare_static_dtype

@ByRef
public TensorIteratorConfig declare_static_dtype(torch.ScalarType dtype)

declare_static_device

@ByRef
public TensorIteratorConfig declare_static_device(@ByVal
                                                         Device device)

declare_static_shape

@ByRef
public TensorIteratorConfig declare_static_shape(@ByVal
                                                        LongArrayRef shape)

declare_static_shape

@ByRef
public TensorIteratorConfig declare_static_shape(@ByVal @Cast(value={"int64_t*","c10::ArrayRef<int64_t>","std::vector<int64_t>&"}) @StdVector(value="int64_t")
                                                        long... shape)

declare_static_shape

@ByRef
public TensorIteratorConfig declare_static_shape(@ByVal
                                                        LongArrayRef shape,
                                                        @ByVal
                                                        LongArrayRef squash_dims)

declare_static_shape

@ByRef
public TensorIteratorConfig declare_static_shape(@ByVal @Cast(value={"int64_t*","c10::ArrayRef<int64_t>","std::vector<int64_t>&"}) @StdVector(value="int64_t")
                                                        long[] shape,
                                                        @ByVal @Cast(value={"int64_t*","c10::ArrayRef<int64_t>","std::vector<int64_t>&"}) @StdVector(value="int64_t")
                                                        long... squash_dims)

build

@ByVal
public TensorIterator build()