[tensor wrapper subclass] Add trace transform for tensor subclasses

[tensor wrapper subclass] Add trace transform for tensor subclasses #5409