التصدير وعقود الأشكال الديناميكية

import torch
from torch import nn
class Scale(nn.Module):
    def forward(self,x):
        return 2*x+1
sample = torch.ones(2,2,dtype=torch.float64,device="cpu")
batch = torch.export.Dim("batch",min=2,max=4)
program = torch.export.export(Scale(),(sample,),dynamic_shapes={"x":{0:batch}},strict=True)
new_input = torch.ones(3,2,dtype=torch.float64,device="cpu")
output = program.module()(new_input)
print("shape", tuple(output.shape))
print("first row", output[0].tolist())

shape (3, 2)
first row [3.0, 3.0]

للدالة العنصرية دفعة ديناميكية محدودة صراحة وعرض سمات ثابت. لم ينفذ التقاط تصدير أو استنتاج قيود أو تشغيل وحدة مصدرة أو حفظ.

def validate(rows):
    if not 2 <= len(rows) <= 4:
        raise ValueError("batch outside 2..4")
    if any(len(row) != 2 for row in rows):
        raise ValueError("expected two features")
    return [[2*value+1 for value in row] for row in rows]
output = validate([[1.0,1.0]]*3)
print("shape", (len(output),len(output[0])))
try:
    validate([[1.0,1.0]]*5)
except ValueError as error:
    print(str(error))

shape (3, 2)
batch outside 2..4

يفحص المدقق المحلي المنفذ مجال شكل وحساباً معلنين، لا حراس torch.export أو مؤثرات ملتقطة أو أثراً مصدراً قابلاً للنقل.