Searched defs:affine_grid_generator (Results 1 – 2 of 2) sorted by relevance
89 Tensor affine_grid_generator(const Tensor& theta, IntArrayRef size, bool align_corners) { in affine_grid_generator() function
4064 def affine_grid_generator(theta: Tensor, size: List[int], align_corners: bool): function