Skip to content

Latest commit

 

History

History
16 lines (13 loc) · 477 Bytes

pytorch.md

File metadata and controls

16 lines (13 loc) · 477 Bytes

troch.topk(input, k, dim = None, largest=True, sorted=True, out=None)

功能: input:张量 k:决定选取k个值 dim:索引维度 返回: Tensor:前k大的值 LongTensor:前k大的值所在的位置

transforms.FiveCrop(size)

transforms.TenCrop(size, vertical_flip=False)

size:所需剪裁图片尺寸

torchvision.utils.make_grid(tensor, nrow=8, padding=2, normalize=False, range=None, scale_each=False, pad_value=0)

功能:制作网格图像