چگونگی حذف حلقه های for در پایتورچ
ده ماه پیش منتشر شده
تعداد بازدید: 176
کد پروژه: 471082
شرح پروژه
import torch
def custom_function(x):
return x ** 2
tensor = torch.tensor([1, 2, 3, 4, 5], device="cuda")
tensor = custom_function(tensor)
print(tensor) # tensor([ 1, 4, 9, 16, 25])
مثلا توی این کد کل آرایه رو به تابع مورد نظر میفرسته و اون تابع روی تک تک درایه های ماتریس اعمال میشه ولی توی این کد پایینی این کار رو نمیتونه انجام بده و ارور میده:
import torch
length = 8
def WToSN(In):
LenSN = 2 ** length
SN = torch.zeros(LenSN, device=torch.device(‘cuda’))
SN[0:In] = 1
return SN
weight = torch.tensor([[1, 2, 3], [4, 5, 6]], device=torch.device(‘cuda’))
print(weight)
WeightSN = WToSN(weight)
print(WeightSN)
i get this error:TypeError: only integer tensors of a single element can be converted to an index
for this line: SN[0:In] = 1
چجوری میتونم این ارور رو برطرف کنم و حلقه for رو حذف کنم؟
ممنون
مهارت ها و تخصص های مورد نیاز
بودجه
5,000 تومان تا 100,000 تومان
مهلت برای انجام
5روز
وضعیت مناقصه
بسته
درباره کارفرما
عضویت ده ماه پیش
