(self, a, axis=None)
| 2776 | ), f"Device discrepancy. First input is on {str(a_device)}, whereas second input is on {str(b_device)}" |
| 2777 | |
| 2778 | def squeeze(self, a, axis=None): |
| 2779 | return cp.squeeze(a, axis=axis) |
| 2780 | |
| 2781 | def bitsize(self, type_as): |
| 2782 | return type_as.itemsize * 8 |