encoderfile 0.4.0-rc.1

Distribute and run transformer encoders with a single file.
Documentation 
1
2
3
4
5
6
7
8
9
10
11

---Generated by Encoderfile ❤️
---Remember: Lua is 1-indexed!

---Postprocess embeddings
---Must return 3D tensor of shape [batch_size, seq_len, *]
---@input Tensor 3D tensor of shape [batch_size, seq_len, hidden_dim]
---@return Tensor
function Postprocess(arr)
    ---Replace with your logic here
    return arr
end