(t, pad, dim=-1, value=0.0)
| 98 | |
| 99 | |
| 100 | def pad_at_dim(t, pad, dim=-1, value=0.0): |
| 101 | dims_from_right = (-dim - 1) if dim < 0 else (t.ndim - dim - 1) |
| 102 | zeros = (0, 0) * dims_from_right |
| 103 | return F.pad(t, (*zeros, *pad), value=value) |
| 104 | |
| 105 | |
| 106 | def pick_video_frame(video, frame_indices): |