python查看数据集的维度、大小_pytorch查看通道数 维数 尺寸大小方式

查看tensor

x.shape # 尺寸

x.size() # 形状

x.ndim # 维数

例如

import torch

parser = argparse.ArgumentParser(description='PyTorch')

parser.add_argument('--img_w', default=144, type=int, metavar='imgw', help='img width')

parser.add_argument('--img_h', default=288, type=int, metavar='imgh', help='img height')

parser.add_argument('--batch-size', default=32, type=int,metavar='B', help='training batch size')

parser.add_argument('--test-batch', default=64, type=int, metavar='tb', help='testing batch size')

class net(nn.Module):

def __init__(self, arch='resnet18'):

super(net, self).__init__()

model_ft = models.resnet50(pretrained=True)

self.visible = model_f