(v)
| 12266 | return n; |
| 12267 | } |
| 12268 | createTupleUnpack(v) { |
| 12269 | const tt = v.type().expect(torch.TupleType); |
| 12270 | const n = this.create('prim::TupleUnpack', [v], 0); |
| 12271 | for (const element of tt.elements()) { |
| 12272 | n.addOutput().setType(element); |
| 12273 | } |
| 12274 | return n; |
| 12275 | } |
| 12276 | createTupleIndex(tup, idx, output_type) { |
| 12277 | const n = this.create('prim::TupleIndex', [tup, idx]); |
| 12278 | n.output().setType(output_type); |