1×80
1×1024
1×1024
1×1024
1×1024
1×seq_len
1×seq_len
1×seq_len
1×512
1×seq_len×512
1×seq_len×128
1×seq_len
1×1024
1×1024
1×1024
1×1024
1×seq_len
1×seq_len
1×seq_len
1×512
1×512
1×512
1×seq_len
1×1024
1×1024
1×1024
1×80
1×1
decoder_input
float32[1,80]
attention_hidden
float32[1,1024]
attention_cell
float32[1,1024]
decoder_hidden
float32[1,1024]
decoder_cell
float32[1,1024]
attention_weights
float32[1,seq_len]
attention_weights_cum
float32[1,seq_len]
attention_context
float32[1,512]
memory
float32[1,seq_len,512]
processed_memory
float32[1,seq_len,128]
mask
boolean[1,seq_len]
MatMul
MatMul_0
float32[80,256]
B
〈80×256〉
Relu
Relu_1
RandomUniform
RandomUniform_2
Cast
Cast_3
Cast
Cast_4
Greater
Greater_6
float64
B
〈〉
Not
Not_7
Cast
Cast_8
Shape
Shape_9
Gather
Gather_11
int64
indices
〈〉
Shape
Shape_12
Gather
Gather_14
int64
indices
〈〉
Unsqueeze
Unsqueeze_15
Unsqueeze
Unsqueeze_16
Concat
Concat_17
Reshape
Reshape_19
int64[1]
shape
〈1〉
Shape
Shape_20
ConstantOfShape
ConstantOfShape_21
Mul
Mul_23
int64
B
〈〉
Equal
Equal_24
Cast
Cast_25
Where
Where_26
Expand
Expand_27
Mul
Mul_28
Mul
Mul_30
float32
B
〈〉
MatMul
MatMul_31
float32[256,256]
B
〈256×256〉
Relu
Relu_32
RandomUniform
RandomUniform_33
Cast
Cast_34
Cast
Cast_35
Greater
Greater_37
float64
B
〈〉
Not
Not_38
Cast
Cast_39
Shape
Shape_40
Gather
Gather_42
int64
indices
〈〉
Shape
Shape_43
Gather
Gather_45
int64
indices
〈〉
Unsqueeze
Unsqueeze_46
Unsqueeze
Unsqueeze_47
Concat
Concat_48
Reshape
Reshape_50
int64[1]
shape
〈1〉
Shape
Shape_51
ConstantOfShape
ConstantOfShape_52
Mul
Mul_54
int64
B
〈〉
Equal
Equal_55
Cast
Cast_56
Where
Where_57
Expand
Expand_58
Mul
Mul_59
Mul
Mul_61
float32
B
〈〉
Concat
Concat_62
Unsqueeze
Unsqueeze_63
Unsqueeze
Unsqueeze_64
Unsqueeze
Unsqueeze_65
LSTM
LSTM_66
float32[1,4096,768]
W
〈1×4096×768〉
float32[1,4096,1024]
R
〈1×4096×1024〉
float32[1,8192]
B
〈1×8192〉
Squeeze
Squeeze_67
Squeeze
Squeeze_68
Unsqueeze
Unsqueeze_69
Unsqueeze
Unsqueeze_70
Concat
Concat_71
Unsqueeze
Unsqueeze_72
MatMul
MatMul_73
float32[1024,128]
B
〈1024×128〉
Conv
Conv_74
float32[32,2,31]
W
〈32×2×31〉
Transpose
Transpose_75
MatMul
MatMul_76
float32[32,128]
B
〈32×128〉
Add
Add_77
Add
Add_78
Tanh
Tanh_79
MatMul
MatMul_80
float32[128,1]
B
〈128×1〉
Squeeze
Squeeze_81
Cast
Cast_82
Where
Where_84
float32
X
〈〉
Softmax
Softmax_85
Unsqueeze
Unsqueeze_86
MatMul
MatMul_87
Squeeze
Squeeze_88
Add
Add_89
Concat
Concat_90
Unsqueeze
Unsqueeze_91
Unsqueeze
Unsqueeze_92
Unsqueeze
Unsqueeze_93
LSTM
LSTM_94
float32[1,4096,1536]
W
〈1×4096×1536〉
float32[1,4096,1024]
R
〈1×4096×1024〉
float32[1,8192]
B
〈1×8192〉
Squeeze
Squeeze_95
Squeeze
Squeeze_96
Concat
Concat_97
Gemm
Gemm_98
float32[80,1536]
B
〈80×1536〉
float32[80]
C
〈80〉
Gemm
Gemm_99
float32[1,1536]
B
〈1×1536〉
float32[1]
C
〈1〉
decoder_output
float32[1,80]
gate_prediction
float32[1,1]
out_attention_hidden
float32[1,1024]
out_attention_cell
float32[1,1024]
out_decoder_hidden
float32[1,1024]
out_decoder_cell
float32[1,1024]
out_attention_weights
float32[1,seq_len]
out_attention_weights_cum
float32[1,seq_len]
out_attention_context
float32[1,512]
×
torch-jit-export
❮
Version
{version}
Copyright ©
Lutz Roeder
Open Model…
.
.
.
OK
≡