mirror of
https://github.com/storytold/storyteller-ml.git
synced 2026-10-12 23:48:57 +00:00
Michael Chung
16 lines
338 B
Python
16 lines
338 B
Python
import torch
|
|
import torch.nn as nn
|
|
# from icefall.utils import make_pad_mask
|
|
|
|
from .symbol_table import SymbolTable
|
|
|
|
# make_pad_mask = make_pad_mask
|
|
SymbolTable = SymbolTable
|
|
|
|
|
|
class Transpose(nn.Identity):
|
|
"""(N, T, D) -> (N, D, T)"""
|
|
|
|
def forward(self, input: torch.Tensor) -> torch.Tensor:
|
|
return input.transpose(1, 2)
|