nn_flatten function

Flattens a contiguous range of dims into a tensor.

torch packageRead PDF manual
  • Maintainer: Daniel Falbel
  • License: MIT + file LICENSE
  • Last published: 2024-05-21

Useful links