View source: R/attention-utils.R
x Tensor [batch, heads, length, depth] x_shape tf$TensorShape of x memory_block_size Integer to dividing length by Return Tensor [batch, heads, length
1 | reshape_by_blocks(x, x_shape, memory_block_size)
|
Add the following code to your website.
For more information on customizing the embed code, read Embedding Snippets.