Args: x: Tensor input tensor [batch, length, features] z1_features: int the number of features of z1 Returns: Tensor, Tensor split tensors [batch, length, z1_features], [batch, length, features-z1_features]
(x: torch.Tensor, z1_features)
Source from the content-addressed store, hash-verified