reshape_by_blocks: Reshape input by splitting length over blocks of...

Description Usage

View source: R/attention-utils.R

Description

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

Usage

1
reshape_by_blocks(x, x_shape, memory_block_size)

ifrit98/transformR documentation built on Nov. 26, 2019, 2:14 a.m.