Module tch_utils::ops_2d

source ·
Expand description

2D Operations

This module contains operations on tensor of shape [N, C, H, W] on each pixel.

Functions

  • dot_product_2d
    Compute the vector dot product between two tensors for each pixel.
  • norm_2d
    Compute the norm of vectors in 2d pictures. Equivalent to linalg_norm but faster.
  • normalize_2d
    Normalize vectors in 2d pictures to unit length.
  • rotate_2d
    Rotate the of vectors in 2d pictures.
  • scale_2d
    Scale the of vectors in 2d pictures.
  • translate_2d
    Translate the of vectors in 2d pictures.