Spaces:
Runtime error
Runtime error
| # -*- coding: utf-8 -*- | |
| # Copyright (c) Facebook, Inc. and its affiliates. | |
| from collections import namedtuple | |
| class ShapeSpec(namedtuple("_ShapeSpec", ["channels", "height", "width", "stride"])): | |
| """ | |
| A simple structure that contains basic shape specification about a tensor. | |
| It is often used as the auxiliary inputs/outputs of models, | |
| to complement the lack of shape inference ability among pytorch modules. | |
| Attributes: | |
| channels: | |
| height: | |
| width: | |
| stride: | |
| """ | |
| def __new__(cls, channels=None, height=None, width=None, stride=None): | |
| return super().__new__(cls, channels, height, width, stride) | |