TensorPlay
API symbolsnn
Copy
View MarkdownDownload .md

tensorplay.nn.functional.affine_grid

tensorplay.nn.functional.affine_grid(theta: TensorBase, size, align_corners=None) → TensorBase[source]

Generate 2D or 3D flow field (sampling grid), given a batch of affine matrices theta.

Port of aten/src/ATen/native/AffineGridGenerator.cpp.

tensorplay.nn.functional.adaptive_max_pool3d_with_indices

Applies a 3D adaptive max pooling over an input signal, returning (output, indices) .

tensorplay.nn.functional.alpha_dropout

Next Page

Ask DeepWiki