org.bytedeco.javacpp

Class tensorflow.MatrixBandPart

java.lang.Object

org.bytedeco.javacpp.Pointer

org.bytedeco.javacpp.tensorflow.MatrixBandPart

All Implemented Interfaces:

AutoCloseable

Enclosing class:

tensorflow

@Namespace(value="tensorflow::ops")
 @NoOffset
public static class tensorflow.MatrixBandPart
extends Pointer

Copy a tensor setting everything outside a central band in each innermost matrix to zero. The band part is computed as follows: Assume input has k dimensions [I, J, K, ..., M, N], then the output is a tensor with the same shape where band[i, j, k, ..., m, n] = in_band(m, n) * input[i, j, k, ..., m, n]. The indicator function in_band(m, n) = (num_lower < 0 || (m-n) <= num_lower)) && (num_upper < 0 || (n-m) <= num_upper). For example:

  # if 'input' is [[ 0,  1,  2, 3]
                   [-1,  0,  1, 2]
                   [-2, -1,  0, 1]
                   [-3, -2, -1, 0]],
 
  tf.matrix_band_part(input, 1, -1) ==> [[ 0,  1,  2, 3]
                                         [-1,  0,  1, 2]
                                         [ 0, -1,  0, 1]
                                         [ 0,  0, -1, 0]],
 
  tf.matrix_band_part(input, 2, 1) ==> [[ 0,  1,  0, 0]
                                        [-1,  0,  1, 0]
                                        [-2, -1,  0, 1]
                                        [ 0, -2, -1, 0]]
  

Useful special cases:

   tf.matrix_band_part(input, 0, -1) ==> Upper triangular part.
   tf.matrix_band_part(input, -1, 0) ==> Lower triangular part.
   tf.matrix_band_part(input, 0, 0) ==> Diagonal.
  

Arguments: * scope: A Scope object * input: Rank k tensor. * num_lower: 0-D tensor. Number of subdiagonals to keep. If negative, keep entire lower triangle. * num_upper: 0-D tensor. Number of superdiagonals to keep. If negative, keep entire upper triangle. Returns: * Output: Rank k tensor of the same shape as input. The extracted banded tensor.

Nested Class Summary

Nested classes/interfaces inherited from class org.bytedeco.javacpp.Pointer

Pointer.CustomDeallocator, Pointer.Deallocator, Pointer.NativeDeallocator

Field Summary

Fields inherited from class org.bytedeco.javacpp.Pointer

address, capacity, limit, position

Constructor Summary

Constructors

Constructor and Description
MatrixBandPart(Pointer p) Pointer cast constructor.
MatrixBandPart(tensorflow.Scope scope, tensorflow.Input input, tensorflow.Input num_lower, tensorflow.Input num_upper)

Method Summary

Modifier and Type	Method and Description
tensorflow.Input	asInput()
tensorflow.Output	asOutput()
tensorflow.Output	band()
tensorflow.MatrixBandPart	band(tensorflow.Output band)
tensorflow.Node	node()
tensorflow.Operation	operation()
tensorflow.MatrixBandPart	operation(tensorflow.Operation operation)

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

Methods inherited from class java.lang.Object

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

Constructor Detail

MatrixBandPart

public MatrixBandPart(Pointer p)

Pointer cast constructor. Invokes Pointer.Pointer(Pointer).

MatrixBandPart

public MatrixBandPart(@Const @ByRef
                      tensorflow.Scope scope,
                      @ByVal
                      tensorflow.Input input,
                      @ByVal
                      tensorflow.Input num_lower,
                      @ByVal
                      tensorflow.Input num_upper)

Method Detail

asOutput

@ByVal
 @Name(value="operator tensorflow::Output")
public tensorflow.Output asOutput()

asInput

@ByVal
 @Name(value="operator tensorflow::Input")
public tensorflow.Input asInput()

node

public tensorflow.Node node()

operation

@ByRef
public tensorflow.Operation operation()

operation

public tensorflow.MatrixBandPart operation(tensorflow.Operation operation)

band

@ByRef
public tensorflow.Output band()

band

public tensorflow.MatrixBandPart band(tensorflow.Output band)