torch_unbind {torch} | R Documentation |
Unbind
torch_unbind(self, dim = 1L)
self |
(Tensor) the tensor to unbind |
dim |
(int) dimension to remove |
Removes a tensor dimension.
Returns a tuple of all slices along a given dimension, already without it.
if (torch_is_installed()) {
torch_unbind(torch_tensor(matrix(1:9, ncol = 3, byrow=TRUE)))
}