powered by
Unflattens a tensor dim expanding it to a desired shape. For use with [nn_sequential.
nn_unflatten(dim, unflattened_size)
Dimension to be unflattened
New shape of the unflattened dimension
if (torch_is_installed()) { input <- torch_randn(2, 50) m <- nn_sequential( nn_linear(50, 50), nn_unflatten(2, c(2, 5, 5)) ) output <- m(input) output$size() }
Run the code above in your browser using DataLab