pygmtools.pytorch_backend.build_batch

pygmtools.pytorch_backend.build_batch(input, return_ori_dim=False)[source]

Pytorch implementation of building a batched tensor