return torch.ops.prims.broadcast_in_dim.default(x, shape, dims).clone() source code is as below: import torch class BroadcastInDim(torch.nn.Module): def forward(self, x): x = x.clone() dims = [0, 1] shape = [x.shape[d] for d in dims] shape.append(1) return torch.ops.prims...