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.