Skip to content

torchrelay.multivers

movedim

movedim(input, source, destination)

squeeze

squeeze(input, dim=None)

unsqueeze

unsqueeze(input, dim)

meshgrid_ij

meshgrid_ij(*tensors)

meshgrid_xy

meshgrid_xy(*tensors)