23 lines
867 B
Python
23 lines
867 B
Python
import torch
|
|
import sys
|
|
|
|
output_file = r"f:\AppsCrucial\ComfyUI_phoenix3\ComfyUI\custom_nodes\ComfyUI-AnotherUtils\scratch\native_nested_check.txt"
|
|
|
|
with open(output_file, "w") as f:
|
|
try:
|
|
# Create a dummy native nested tensor
|
|
v = torch.randn((1, 12, 10, 32, 32))
|
|
a = torch.randn((1, 8, 10, 16))
|
|
# Note: torch.nested_tensor requires same dims usually, but let's see
|
|
try:
|
|
nt = torch.nested.nested_tensor([v, a])
|
|
f.write(f"Native NT created. is_nested: {nt.is_nested}\n")
|
|
f.write(f"Shape: {nt.shape}\n")
|
|
except Exception as e:
|
|
f.write(f"Native NT creation failed: {e}\n")
|
|
|
|
# Check if samples.is_nested exists on standard tensors
|
|
f.write(f"Standard tensor is_nested: {v.is_nested}\n")
|
|
except Exception as e:
|
|
f.write(f"Error: {e}")
|