Source code for hyperion.torch.transforms.reshape

"""
 Copyright 2019 Johns Hopkins University  (Author: Jesus Villalba)
 Apache 2.0  (http://www.apache.org/licenses/LICENSE-2.0)
"""

import torch


[docs]class Reshape(object):
[docs]    def __init__(self, shape):
        self.shape = shape

    def __call__(self, x):
        return torch.reshape(x, shape=self.shape)