getTensor

abstract fun Tensor<T>.getTensor(i: Int): Tensor<T>(source)

Returns the tensor at index i For more information: https://pytorch.org/cppdocs/notes/tensor_indexing.html

Return

subtensor of the original tensor with index i

Parameters

i

index of the extractable tensor


open fun Tensor<T>.getTensor(first: Int, second: Int): Tensor<T>(source)