IntTensor

class IntTensor

An int tensor, which stores its underlying data in an int array and tracks the strides needed to access data for each dimension.

Types

Companion

object Companion

Functions

fun at(pos: Int): Int

flatten

fun flatten(startDim: Int = 0, endDim: Int = rank - 1): IntTensor

get

operator fun get(index: Int): IntTensor

normalize

open fun normalize(): IntTensor

Return an equivalent tensor whose representation is an IntTensor with natural layout and zero offset (that is, with a contiguous data representation).

reshape

fun reshape(newShape: Shape): IntTensor

view

fun view(index: Int, axis: Int): IntTensor

Properties

dataIterator

val dataIterator: Iterable<Int>

rank

val rank: Int

shape

val shape: Shape

size

val size: Int