2021年3月16日
PyTorch——卷积（一）
摘要： 1 torch.nn.Conv2d(in_channels, out_channels, kernel_size, stride=1, padding=0, dilation=1, groups=1, bias=True, padding_mode='zeros') Parameters 1 in_ 阅读全文
posted @ 2021-03-16 14:57 一剑光寒十四州 阅读(125) 评论(0) 推荐(0)
Tensor——各个机器学习框架下Tensor中的通道顺序(一)
摘要： 序号 框架 通道顺序 1 Caffe NCHW 2 Tensorflow 默认NHWC,也支持NCHW 3 PyTorch NCHW 4 TensorRT NCHW N：batch C：channel H：height W：width 阅读全文
posted @ 2021-03-16 14:28 一剑光寒十四州 阅读(297) 评论(0) 推荐(0)