docstr: Test behavior with a Tensor input
docstr_tok: ["Test","behavior","with","a","Tensor","input"]
code: def test_get_tensor_size_bytes_map_with_tensor_input(self) -> None:
      """Test behavior with a Tensor input"""
  
      inputs = torch.rand(1, 2, 3)
      tensor_map = get_tensor_size_bytes_map(inputs)
      self.assertEqual(len(tensor_map), 1)
      self.assertTrue(inputs in tensor_map)
      self.assertEqual(
          tensor_map[inputs],
          inputs.size().numel() * inputs.element_size(),
      )
  
imports: import time
  import unittest
  from collections import deque, namedtuple
  from dataclasses import dataclass
  from typing import cast, Deque, List
  import torch
  from torchtnt.utils.memory import get_tensor_size_bytes_map, measure_rss_deltas, RSSProfiler
label_desc: Use image node_id 1 for calling the MemoryTest obj's underlying member method code with example usage: obj.test_get_tensor_size_bytes_map_with_tensor_input() without return types