Args: shape (list of int): Specified shape for output. At most one dimension of the new shape can be -1. In this case, the value is inferred from the size of the tensor and the remaining dimensions. A dimension could also be 0,
(self, shape)
| 733 | """ |
| 734 | |
| 735 | def __init__(self, shape): |
| 736 | """ |
| 737 | Args: |
| 738 | shape (list of int): Specified shape for output. At most one |
| 739 | dimension of the new shape can be -1. In this case, the |
| 740 | value is inferred from the size of the tensor and the |
| 741 | remaining dimensions. A dimension could also be 0, |
| 742 | in which case the actual dimension value is unchanged |
| 743 | (i.e. taken from the input tensor). |
| 744 | """ |
| 745 | super(Reshape, self).__init__() |
| 746 | self.shape = shape |
| 747 | |
| 748 | def forward(self, x): |
| 749 | """ |