Source code for hyperion.torch.narchs.net_arch

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

import numpy as np

import torch.nn as nn

from ..torch_model import TorchModel


[docs]class NetArch(TorchModel):
[docs]    def in_context(self):
        return 0

[docs]    def in_dim(self):
        return len(self.in_shape())

[docs]    def out_dim(self):
        return len(self.out_shape())

[docs]    def in_shape(self):
        raise NotImplementedError()

[docs]    def out_shape(self, in_shape=None):
        raise NotImplementedError()