Module tch_utils::tensor_init
source · Expand description
Tensor initialization functions
This module contains functions to initialize tensors that are not in the tch create.
Functions
- Generate a tensor that contains a vec2 from (-1, -1) to (1, 1) depending on the position.