(Tensor) the input tensor.
(int) k for the k-th smallest element
(int, optional) the dimension to find the kth value along
(bool) whether the output tensor has
Returns a namedtuple
(values, indices) where
values is the
smallest element of each row of the
input tensor in the given dimension
indices is the index location of each element found.
dim is not given, the last dimension of the
input is chosen.
TRUE, both the
are the same size as
input, except in the dimension
they are of size 1. Otherwise,
dim is squeezed
torch_squeeze), resulting in both the
indices tensors having 1 fewer dimension than the
1 2 3 4 5 6 7 8 9
Add the following code to your website.
For more information on customizing the embed code, read Embedding Snippets.