(feats)
| 636 | # shaped like: [ (batch_agent1, batch_agent2, ...) , channel, h, w] |
| 637 | @staticmethod |
| 638 | def agents_to_batch(feats): |
| 639 | agent_num = feats.shape[1] |
| 640 | feat_list = [] |
| 641 | for i in range(agent_num): |
| 642 | feat_list.append(feats[:, i, :, :, :]) |
| 643 | feat_mat = torch.cat(tuple(feat_list), 0) |
| 644 | return feat_mat |
| 645 | |
| 646 | def forward_encoder(self, x1, x_next, mask_ratio): |
| 647 | # embed patches |
nothing calls this directly
no outgoing calls
no test coverage detected